问题描述:
给定一个长度为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
输出范例 :
6
思路:
这题也是学习的思路,用sum记录下前n项和,mod记下前n项和的不同余数的个数。
如果前 i 项和前 j 项的和相同,那么它们的差就可以构成一段连续序列且除以K的余数为0
因此再计算每项的排列组合数目(C上2下该余数的个数)
踩坑点:
1、没注意题目的数量级,最高数字可达100000*100000,超出了int的上限,需要把sum,res都改成longlong才行,否则会丢结果
2、一开始把余数为0的单独拿出来处理,没有进行排列组合计算,事实上余数为0的也要进行排列组合计算,并且最后要把余数为0的单独加一次
代码实现:
#include<bits/stdc++.h>
using namespace std;
int main()
{
int N,K;
long long res=0;
cin>>N>>K;
int *arr=new int[N];
//mod[i]代表前i个元素的最多有多少个k倍区间
long long *mod=new long long[K];
//sum[i]代表前i+1个元素和
long long *sum=new long long[N];
fill(sum,sum+N,0);
fill(mod,mod+K,0);
for(int i=0;i<N;i++)
{
cin>>arr[i];
if(i!=0)
sum[i]=sum[i-1]+arr[i];
else
sum[i]=arr[i];
mod[sum[i]%K]++;
}
for(int i=0;i<K;i++)
{
if(mod[i]>0) res+=mod[i]*(mod[i]-1)/2;
}
res+=mod[0];
cout<<res<<endl;
}