k倍区间(前缀和)

给定一个长度为 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

解题思路:打眼一看就是前缀和,然后就就不会了

经过看了一桶题解后 发现Ai,Ai+1,…Aj 之和是 K 的倍数
嗷咬 ,这里可以优化
由前缀和(s[i]-s[j]) 是k的倍数可推出 s[i]%k=s[j]%k
这样的话只需找到 这样的s[i]和s[j] (s[i]%k=s[j]%k) 就ok了

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
const int N = 100010;

typedef long long LL;

LL cnt[N],s[N];
LL res;

int n,k;

int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>s[i];
        s[i]+=s[i-1];
    }
    
    cnt[0]=1;   //   考虑一个点本身可以%k=0   
                //  单个序列取余=0也构成K倍区间
    for(int i=1;i<=n;i++)
    {
        res+=cnt[s[i]%k];
        cnt[s[i]%k]++;
    }
    
    cout<<res<<endl;
    return 0;
}