AcWing每日一题(提高组)--K倍区间

https://www.acwing.com/problem/content/1232/

求出一个数组中是k的倍数的连续子序列有多少个。

因为数组元素大于0,所以保证前缀和数组是一个单调上升数组。

将前缀和数组分为k类,模k为0,模k为1...

对于每一个前缀和只需要找和他同类的有多少个就可以了。

因为二者如果同类,那么s[ i ] - s[ j ] 一定是k的倍数。

 1 #include<iostream>
 2 using namespace std;
 3 typedef long long LL;
 4 const int N=100010;
 5 LL s[N],cnt[N];
 6 int main(void){
 7     int n,k;
 8     cin>>n>>k;
 9     for(int i=1;i<=n;i++){
10         cin>>s[i];
11         s[i]+=s[i-1];
12     }
13     LL res=0;
14     for(int i=0;i<=n;i++){//从0开始,使得mod k=0的类数量为1
15         res+=cnt[s[i]%k];
16         cnt[s[i]%k]++;
17     }
18     cout<<res;
19     return 0;
20 }

 

posted on 2021-02-09 22:44  greenofyu  阅读(45)  评论(0编辑  收藏  举报