算法：k倍区间（前缀和）

思路

利用前缀和可以快速计算出一个区间的值，然后利用前缀和的取余相同时，相减之后就会等于0，说明如果有k个具有相同余数的区间的话，就能有(k-1)+(k-2)+…+1个区间满足的情况。cnt[0]++是因为如果s[i]取余的值为0时，就相当于从总区间的左端点到i的区间满足情况，所以应该提前加上cnt[0]++。

问题描述

给定一个长度为 N 的数列，A1,A2,…AN，如果其中一段连续的子序列 Ai,Ai+1,…Aj 之和是 K 的倍数，我们就称这个区间 [i,j] 是 K 倍区间。

你能求出数列中总共有多少个 K 倍区间吗？

输入格式

第一行包含两个整数 N 和 K。

以下 N 行每行包含一个整数 Ai。

输出格式

输出一个整数，代表 K 倍区间的数目。

数据范围

1≤N,K≤100000,
1≤Ai≤100000
输入样例：
5 2
1
2
3
4
5
输出样例：
6

代码

#include<iostream>
using namespace std;
const int N = 100010;
typedef long long LL;
LL s[N], cnt[N];
int main(){
    int n, k;
    scanf("%d%d", &n, &k);
    for (int i = 1; i <= n; ++i){
        scanf("%lld", &s[i]);
        s[i] += s[i-1];
    }
    LL ans = 0;
    cnt[0]++;
    for(int i = 1; i <= n; ++i){
        ans += cnt[s[i] % k];
        cnt[s[i] % k]++;
    }
    cout << ans << endl;
}

原题链接