2017第八届蓝桥杯省赛-大学B组 k倍区间(数论)

标题： 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;
}