蓝桥杯每日一题（前缀和）

K倍区间

给定一个长度为 N 的数列，$A_1,A_2,\dots A_N$如果其中一段连续的子序列 $A_i,A_{i+1},\dots A_j$ 之和是 K 的倍数，我们就称这个区间 $[i,j]$是 K 倍区间。

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

输入格式

第一行包含两个整数 N 和 K。

以下 N 行每行包含一个整数 $A_i$。

输出格式

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

数据范围

1≤N,K≤100000
1≤Ai≤100000

输入样例：

5 2
1
2
3
4
5

输出样例：

6

//对于前缀和，符合要求的是(s[r]-s[l-1])%k==0 --> s[r]%k==s[l-1]%k
//即对于每一个前缀和取余k的值存入哈希表，查找当前位置前面的取余k相同的个数，加上即可
//注意一开始cnt[0]++,如果第一次出现的前缀和取余等于0，则不需要寻找前面相同的取余数也要加1
#include<iostream>
using namespace std;
int n,k;
const int N=1e5+10;
typedef long long ll;
ll ans;
ll s[N],cnt[N];
int main(){
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>s[i];
        s[i]+=s[i-1];
    }
    cnt[0]++;//如果第一次出现的前缀和取余等于0，则不需要寻找前面相同的取余数也要加1
    for(int i=1;i<=n;i++){
        ans+=cnt[s[i]%k];
        cnt[s[i]%k]++;
    }
    cout<<ans<<endl;
    return 0;
}