Description
给你一个长度为N的正整数序列,如果一个连续的子序列,子序列的和能够被K整除,那么就视此子序列合法,求原序列包括多少个合法的连续子序列?对于一个长度为8的序列,K=4的情况:2, 1, 2, 1, 1, 2, 1, 2 。它的答案为6,子序列是位置1->位置8,2->4,2->7,3->5,4->6,5->7。
Input
第一行:T,表示数据组数
对于每组数据:
第一行:2个数,K,N
第二行:N个数,表示这个序列
Output
共T行,每行一个数表示答案
Sample Input
2 7 3 1 2 3 4 8 2 1 2 1 1 2 1 2
Sample Output
0 6
Data Constraint
Hint
100%数据满足
1<=T<=20
1<=N<=50000
1<=K<=1000000
序列的每个数<=1000000000
30%数据满足
1<=T<=10
1<=N,K<=1000
正解
考虑使用 前缀和 和 桶 来解题。
做一遍前缀和,并将每个位置的前缀和 mod k 后按余数存入 桶 中。我们知道当两个位置的前缀和 mod k 的余数相同时,这两个位置的区间和就是 k 的余数。所以,当桶内的一个位置的值大于1时,那么就给答案加上 C^{2}_{a_{i}}Cai2
,最后特判一下是否整个序列是否符合要求,是的话再给答案加 1 ,最后输出。
时间复杂度 O(Tn^{2})O(Tn2)
,具体的细节见 代码 部分。
(此题解参考WZX dalao)
CODE
var
t,n,k,x,ans,mo:int64;
i,j:longint;
a:array[0..1000000]of int64;
begin
readln(t);
for j:=1 to t do begin
fillchar(a,sizeof(a),0);
readln(k,n);
mo:=0;
for i:=1 to n do begin
read(x);
mo:=(mo+x) mod k;
inc(a[mo]);
end;
ans:=a[0];
for i:=0 to k-1 do
ans:=ans+a[i]*(a[i]-1) div 2;
writeln(ans);
end;
end.