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

分析:

首先我们要知道前缀和是什么意思，就是前面几个数的和。区间a，和区间b，已知a<b，suma，sumb分别为a，b的前缀和（从序列第一个数到它自身的所有数之和，如：3的前缀和：1+2+3，5
的前缀和：1+2+3+4+5），且suma%k=x，sumb%k=x。那么可得（sumb-suma）%k=0。将前面相同的抵消即为：b-a区间，即由b-a产生的区间c，一定为一个k倍区间，因为可以整除。

那么我们就可以将前缀和模k的值为0，1，2…k-1的区间数分别求出来，然后分别计算。例如前缀和模k为1的区间有x个。那么就可以产生C(x,2)个k倍区间（即x(x-1)/2个，从x个中任意选择两个）。然后对0~k-1分别进行此操作并求和。
需要注意的是还需在最后加上那些前缀和模k为0的区间。（因为其区间本身就是k倍区间如：[1 2 3],[1 2 3 4]）

附上代码：

/*

输入：
5 2
1 2 3 4 5
程序应该输出：
6

*/ 
#include<iostream>
using namespace std;
typedef long long ll;
int main() {
	int n, k;
	int a[100000] = {0};
	int ans[100000] = {0};
	ll sum = 0,count =0;
	cin >> n >> k;
	for (int i = 0; i < n; i++) {
		cin >> a[i];
		sum += a[i];
		ans[sum % k]++;//前缀和相同,开始计数
	}
	//如果有k个前缀和相同的任意选两个值为：x*(x-1)/2个k倍区间
	for (int i = 0	; i < k; i++) {
		count += ans[i] * (ans[i] - 1) / 2;
	}
//当为ans[0]时他自己本身就是k倍区间，如：1 2 3和1 2 3 4,加上即可
	count += ans[0];
	cout << count << endl;
	return 0;
}

运行结果：