蓝桥杯2017 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

思路：

1.暴力法
依次穷举区间和，再对和取余数。

#include<iostream>

using namespace std;

int arr[100010];

int main() {
	int N, K;
	cin >> N >> K;
	for (int i = 0; i < N; i++) {
		cin >> arr[i];
	}
	long long sum ;
	int cnt = 0;
	for (int i = 0; i < N; i++) {
		sum = 0;
		for (int j = i; j < N; j++) {
			sum += arr[i];
			if (sum % K == 0)
				cnt++;
		}
	}
	cout << cnt;
}

2.前缀和（用于静态数组的区间和）

开辟一个数组sum[n]用于放sum[i]=∑a[i]，区间[i,j]的和就是sum[j]-sum[i-1]。和第一种方法差别不大。

#include<iostream>

using namespace std;

int arr[100010];
int sum[100010];

int main() {
	int N, K;
	cin >> N >> K;
	sum[0] = 0;
	for (int i = 1; i < N; i++) {
		cin >> arr[i];
		sum[i] = sum[i - 1] + arr[i];
	}
	int cnt = 0;
	for (int i = 1; i < N; i++) {
		for (int j = i; j < N; j++) {
			if ((sum[j]-sum[i-1]) % K == 0)
				cnt++;
		}
	}
	cout << cnt;
}

3.前缀和+同余方程的结论

同余方程的结论：

①如果(a+b) mod k == 0，那么(a mod k + b) mod k== 0;
②如果a mod k == b mod k ==0 ，那么(b-a) mod k == 0；

那么(sum[i-1]+a[i]) mod k就可以转化为(sum[i-1] mod k + a[i]) mod k
在求sum数组时就进行取余数，得到sum[i]=(sum[i-1]+a[i]) mod k。然后利用结论②，可以得到如果[i,x]区间的和和[i,j]区间的和都可以被k整除，[x,j]的区间的数的和也可以被k整除。就将问题转化为了sum中0的全排列问题。

#include<iostream>

using namespace std;

int arr[100010];
int sum[100010];

int main() {
	int N, K;
	cin >> N >> K;
	sum[0] = 0;
	int cnt = 0;
	for (int i = 1; i < N; i++) {
		cin >> arr[i];
		sum[i] = (sum[i - 1] + arr[i])%K;
		if (sum[i] == 0)
			cnt++;
	}
	cout << cnt*(cnt-1)/2;//全排列公式
}