2017年省赛 k倍区间（前缀和）

·原题链接：用户登录

题目描述

给定一个长度为 N 的数列，A1⋯AN​，如果其中一段连续的子序列 Ai,⋯Aj​ ( i \leq ji≤j ) 之和是 KK 的倍数，我们就称这个区间 [i, j] 是 K 倍区间。

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

输入描述

第一行包含两个整数 N 和 K( 1≤N,K≤1e5 )。

以下 N 行每行包含一个整数 A_iAi​ ( 1 \leq A_i \leq 10^51≤Ai​≤105 )

输出描述

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

输入输出样例

示例

输入

5 2
1
2
3
4
5

输出

6

运行限制

• 最大运行时间：2s
• 最大运行内存: 256M

【代码及思路】

#include<iostream>
using namespace std;
typedef long long ll;
const int N=1e5;
ll a[N];
int main()
{
    int n,k,x;
    ll sum=0;
    cin>>n>>k;
    for(int i=0;i<n;i++)
    {
        cin>>x;
        sum+=x;
        a[sum%k]++;//前缀和模k值相同的放在一组 
    }
    ll ans=0;
    for(int i=0;i<k;i++)
    {
        ans+=a[i]*(a[i]-1)/2;
        /*从前缀和模k值相等的区间中任选两个
        且所选的区间不能重复，小在前，大在后
        所以不涉及排序，选出的结果要除以2 
        */ 
    }
    ans+=a[0];
    cout<<ans<<endl;
    return 0;
}