第八届蓝桥杯 K倍区间

标题： k倍区间

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

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

输入
-----
第一行包含两个整数N和K。(1 <= N, K <= 100000)  
以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)  

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

例如，
输入：
5 2
1  
2  
3  
4  
5  

程序应该输出：
6

---

原先想过是否可以用尺取法，但是区间数的加和不满足单调性，因此无法使用尺取法

emm，拿到题的第二反应就是把它划分成同余类，但是接下来怎么优化也没啥想法学习了一个比较巧妙的想法，区间和肯定是要用前缀和处理的，可以通过找在自己之前有几个数和自己mod k同余来求解

讲的很清楚的参考博客

学到了：如果要在一堆数中找到两两相同的个数（有点生日攻击(╥﹏╥)的赶脚），可以开个数组记录在自己之前和自己一样的数的个数，直接在答案上加上去 （本题有序） 感觉这个思路还是很巧妙的，从把问题转化为mod k同余，再到找个数

#include<iostream>
using namespace std;
const int N=1e5+10;
typedef long long LL;
LL a[N];
LL sum[N];
LL cnt[N];             //cnt[i]表示在当前位置之前有几个数和自己mod k同余
int n,k;
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;++i)
        cin>>a[i];
    sum[1]=a[1]%k;
    for(int i=2;i<=n;++i)
        sum[i]=(a[i]%k+sum[i-1])%k;
    LL ans=0;
    for(int i=1;i<=n;++i)
        ans+=cnt[sum[i]]++;         //注意要自加
    ans+=cnt[0];                    //注意单个数也可以看做一个k倍区间
    cout<<ans<<endl;

}