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

解题思路：本题嘛，我用sum[i]来存前 i 个元素的和mod k 的结果，然后呢f[ i ] 表示sum数组中有多少个元素值是 i 。很显然，当sum[ i ] 为 0 时，代表[0 , i ] 为 k 的倍数。另外的，当且仅当sum[ i ] == sum[ j ] 时，[ i , j ]才是 k 的倍数。（证明 ? 显而易见嘛。）所以最终答案就是f[ 0 ] + f[ i ] 的排列数，0 <= i <= k-1。

#include<cstring>
#include<cstdio>
#include<iostream>
using namespace std;
int n,k;
const int maxn = 100100;
int f[maxn];
int a[maxn];
int sum[maxn];
long long jie(int x){
	long long res = (1+x)*x/2;
	return res;
}
void solve(){
	long long ans = 0;
	for(int i = 1;i <= n;i++) 
		sum[i] = (sum[i-1]+a[i]) % k;
	for(int i = 1;i <= n;i++) f[sum[i]]++;
	ans += f[0];
	for(int i = 0;i < k;i++){
		ans += jie(f[i] - 1);
	}
	printf("%lld\n",ans);
}
int main(){
	scanf("%d%d",&n,&k);
	for(int i = 1;i <= n;i++) scanf("%d",a+i);
	solve();
	return 0;
}