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