资源限制
内存限制:256.0MB C/C++时间限制:1.0s Java时间限制:3.0s Python时间限制:5.0s
给定一个长度为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
资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗 < 2000ms
注:1.本题如果用暴力枚举,会显示运行超时,无法通过所有测试点。
解决方法:通过前i-1项的和加上第i项的数字,得到前i项的和 ,可以减少一些时间。然后对k取余,余数为0的则为k的倍数;或者余数相同的做差后也是k的倍数 。将俩个相加后即为最后的答案。
2.在我做本题时,第三个测试点卡了好久,最后才试出来需要给count进行强制转换(倒数第五行)
#include<bits/stdc++.h>
using namespace std;
int a[100001], sum[100001], num[100001]={0};
int main(){
long long int n, i, count=0, k, j, x;
scanf("%lld %lld", &n, &k);
sum[0]=0;
for(i=1;i<=n;i++){
scanf("%lld", &a[i]);
sum[i]=(sum[i-1]+a[i])%k;//通过前i-1项的和加上第i项的数字,得到前i项的和对k取余的值
num[sum[i]]++;
}
count+=num[0];//余数为0的说明是K的倍数
for(i=0;i<k;i++){
//printf("%d\n", num[i]);
count+=(long long)num[i]*(num[i]-1)/2;//利用同余做差是模的倍数
}
printf("%lld", count);
return 0;
}