k倍区间 | 前缀和

k倍区间 - 蓝桥云课 (lanqiao.cn)

 

 

#include<iostream>
using namespace std;
#define ios_base \
    ios::sync_with_stdio(false);\
    cin.tie(nullptr);\
    cout.tie(nullptr)
const int N = 1e5+10;
int n,k;//cnt[]用来计数不同余数的出现次数
long long a[N],cnt[N];//cnt[]的下标可不只有0~9这么简单，举个例子：999%9999，保险起见还是开long long cnt[],而a[]要作为前缀和数组所以也要开大点
long long res=0;
int main()
{
    ios_base;
    cin>>n>>k;
    for (int i = 1; i <= n; i++)
    {
        cin>>a[i];
        a[i]+=a[i-1];//构建前缀和数组
    }
    cnt[0]++;
    for (int i = 1; i <= n; i++)
    {
        res+=cnt[a[i]%k];
        cnt[a[i]%k]++;
    }
    cout<<res<<'\n';

    return 0;
}