jzoj 2644. 【NOIP2011模拟10.31】数列

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}}Cai​2​ ，最后特判一下是否整个序列是否符合要求，是的话再给答案加 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.