标题: k倍区间
给定一个长度为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
题解:这道题。看到之后,我的第一反应就是,居然这么简单?,暴力直接来了一发。然后就没看这道题了。后来我又仔细看了看。这道题数据有点小坑。直接暴力求解枚举各个区间一定会超时。然后我就想了想。连续区间和,果然用前缀和最为方便。然后又自信提交了一发。后来再看的时候发现。我发现(a[l] - a[r])% k = 0 是我们的目标(a[i]代表前缀和),稍微变形一下。就是 a[l] % k = a[r] % k。也就是说。如果我们在处理前缀和区间的时候直接取模k,然后前缀和相等的a[i]相减就是一个我们要找的区间。这就意味着,我们只要通过桶排序,来记录有多少个相等a【i】。 若有相等的a[i]有n个,那么就可以看成从n个物品里面选两个有多少种选法。
注意:因为0这个区间是要计算进去的。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 100000+5;
int a[N];
int sum[N];
int main(){
int n,k;
cin >> n >>k ;
for(int i = 1 ; i <= n ; i ++){
int x;
cin >> x;
a[i] = a[i - 1] + x;
a[i] %= k;
}
sum[0] = 1;
int ans = 0;
for(int i = 1 ; i <= n; i ++){
sum[a[i]]++;
}
for(int i = 0 ; i < n ; i++){
ans += sum[i]*(sum[i]-1)/2;
}
cout << ans << endl;
}