#P16269. [蓝桥杯 2026 省 Java B 组] 量子态叠加计数器

[蓝桥杯 2026 省 Java B 组] 量子态叠加计数器

题目描述

某量子实验室记录了 NN 个量子比特在第 1,2,…,T1, 2, \dots, T 个时刻的状态。每个状态值为 00 或 11。

对任意一个量子比特和任意一个时刻区间 [L,R][L, R](1≤L≤R≤T1 \leq L \leq R \leq T),若该量子比特在区间内状态为 11 的次数恰好为 KK,则称该量子比特在该区间产生了一次有效叠加。

现在,请你统计:在所有量子比特、所有时刻区间中,有效叠加出现的总次数。

输入格式

第一行包含三个整数 N,T,KN, T, K,分别表示量子比特数量、时刻数量、目标次数。

接下来 NN 行,每行包含 TT 个整数(00 或 11),其中第 ii 行表示第 ii 个量子比特在各时刻的状态序列。

输出格式

输出一行,一个整数,表示有效叠加的总次数。

3 5 2
1 0 1 0 1
0 1 1 0 0
1 1 0 0 1
14
2 4 1
1 0 0 1
0 0 0 0
6
1 6 3
1 1 0 1 1 0
3
1 3 0
0 0 0
6

提示

【样例说明 1】

对于第 11 个量子比特 1 0 1 0 11\ 0\ 1\ 0\ 1,满足条件的区间有:[1,3][1,3]、[1,4][1,4]、[2,5][2,5]、[3,5][3,5]。共 44 个。

对于第 22 个量子比特 0 1 1 0 00\ 1\ 1\ 0\ 0,满足条件的区间有:[1,3][1,3]、[1,4][1,4]、[1,5][1,5]、[2,3][2,3]、[2,4][2,4]、[2,5][2,5]。共 66 个。

对于第 33 个量子比特 1 1 0 0 11\ 1\ 0\ 0\ 1,满足条件的区间有:[1,2][1,2]、[1,3][1,3]、[1,4][1,4]、[2,5][2,5]。共 44 个。

因此总次数为 4+6+4=144 + 6 + 4 = 14。

【样例说明 2】

对于第 11 个量子比特 1 0 0 11\ 0\ 0\ 1,恰好包含 11 个状态为 11 的区间有:[1,1][1,1]、[1,2][1,2]、[1,3][1,3]、[2,4][2,4]、[3,4][3,4]、[4,4][4,4]。共 66 个。

对于第 22 个量子比特 0 0 0 00\ 0\ 0\ 0,任意区间内都没有状态为 11,因此不存在恰好包含 11 个状态为 11 的区间。

所以答案为 6+0=66 + 0 = 6。

【样例说明 3】

唯一的量子比特为 1 1 0 1 1 01\ 1\ 0\ 1\ 1\ 0。

恰好包含 33 个状态为 11 的区间有:[1,4][1,4]、[2,5][2,5]、[2,6][2,6]。共 33 个。

【样例说明 4】

唯一的量子比特在所有时刻的状态都为 00。

当 K=0K = 0 时,需要统计区间内恰好有 00 个状态为 11 的情况,也就是区间内所有值都为 00 的区间。

当 T=3T = 3 时,一共有 3×42=6\frac{3 \times 4}{2} = 6 个区间,且它们全部满足条件,因此答案为 66。

【评测用例规模与约定】

对于 30%30\% 的评测用例,N≤10N \leq 10,T≤100T \leq 100;

对于 60%60\% 的评测用例,N≤50N \leq 50,T≤500T \leq 500;

对于所有评测用例,1≤N≤2001 \leq N \leq 200,1≤T≤10001 \leq T \leq 1000,0≤K≤T0 \leq K \leq T;保证输入中的所有状态值均为 00 或 11。