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 }