蓝桥杯 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 

1.首先明白什么是k倍区间：求区间[l,r]的和是k的倍数的个数 。

2.方法思路：通过前缀和来求出，我们规定sum[i]表示第1个元素到第i个元素的和，那么sum[r] - sum[l-1]就是区间[l,r]的和。区间[l,r]的和是k的                          倍数即(sum[r] - sum[l-1])%k == 0 即sum[r]%k == sum[l-1]%k。

3.具体内容：

       a.对前1个数的和1取模为1， 个数+0
　　b.对前2个数的和3取模为1， 个数+1。为什么这里要加1，因为对a执行后，最后取模为1.
　　c.对前3个数的和6取模为0， 个数+0。因为对a,b操作后，没有值为0的。
　　e.对前4个数的和10取模为0，个数+1。因为c操作是0.
　　d.对钱5个数的和15取模为1，个数+2。因为a,b是1，所以要加2.

4.只要最后对前缀和取余是0,本身就是一个k倍区间，所以最后要加上count[0].

#include<bits/stdc++.h>
using namespace std;
long long cnt[1000005]={0};
long long a[1000005];
int main()
{
    long long n,k,num =0;
    cin>>n>>k;
    cin>>a[0];
    a[0] %= k;
    cnt[a[0]]++;
    for(int i=1;i<n;i++)
    {
        cin>>a[i];
        a[i] = (a[i-1]+a[i])%k;
        //cout<<"arr["<<i<<"]="<<arr[i]<<endl;
        num += cnt[a[i]]++;
    }
    cout<<num+cnt[0]<<endl;//加起来本身是偶数，它就是一个区间，所以最后需要再加上。 
    return 0;


}