题目描述:
给定一个长度为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=(sum+x)%k;
//其实这个sum不是前缀和,是前缀和%k的结果,这样是的sum的取值区间落在【0,k】。
//这样,sum就可以不用lang lang 了。
mod[100005]={0};
//用来存放sum%k同模的个数,全部初始化为0。
for(int i=0;i<k;i++)
{
ans+=(mod[i]*(mod[i]-1))/2;
}
//组合数,从mod[i]个中选两个。
//解释:
//假设有n个前缀和,它们同模,模k的结果都是x (0=<x<k),
//那么这n个数中去掉任意两个,其对应的输入样例中的区间和一定是k的倍数,
//所以,模为x,满足题目要求的区间数就是从mod[x]中选任意两个。
ans=mod[0];
/*其实,这个操作,
for(int i=0;i<k;i++)
{
ans+=(mod[i]*(mod[i]-1))/2;
}
*/并不能算出全部个数。
//刚才有段解释:
//组合数,从mod[i]个中选两个。
//解释:
//假设有n个前缀和,它们同模,模k的结果都是x (0=<x<k),
//那么这n个数中去掉任意两个,其对应的输入样例中的区间和一定是k的倍数,
//所以,模为x,满足题目要求的区间数就是从mod[x]中选任意两个。
//这段解释是说去掉其中的两个,那么,模k等于0,是否有其他区间可能?
//如果出现某个前缀和模k等于0,
//那么从输入样例的第一个数,到使得前缀和模k等于0的那一个输入值(包括),
//这个区间本身就是一个k倍区间。
//所以,出现mod[0]个前缀和模k等于0的情况,就会产生mod[0]个k倍区间
完整代码:
#include<iostream>
#include<string.h>
using namespace std;
int main()
{
int n,k,x,sum=0;
cin>>n>>k;
long long ans=0,mod[100005]={0};
while(n--)
{
cin>>x;
sum=(sum+x)%k;
mod[sum]++;
}
ans=mod[0];
for(int i=0;i<k;i++)
{
ans+=(mod[i]*(mod[i]-1))/2;
}
cout<<ans<<endl;
return 0;
}