5 2 1 2 3 4 5
6
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 100010;
#define ll long long
ll n, k,t, q[maxn],cnt[maxn],qian=0;
int main() {
scanf("%lld%lld", &n, &k);
for (int i = 1; i <= n; i++) { //从1开始,方便后边处理
scanf("%lld", &t);
qian = (qian + t); //求输入数组前缀
q[qian%k]++; //求前缀和的取余,并用桶排序、统计个数
}
q[0]++; //注意 q[0]=0,所以要自加,否则f[1]会没法判读
ll flag = 0;
//for (int i = 0; i <k; i++)
// cout << q[i] << " ";
for (int i = 0; i < k; i++) {
if(q[i])
flag += (q[i] * (q[i] - 1) / 2); //要注意这个公式,n个数,从里边取两个,共有n*(n-1)/2种方法
}
cout << flag << "\n";
return 0;
}
原文:https://www.cnblogs.com/52dxer/p/10556722.html