历届试题 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




　　请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。


　　注意：
　　main函数需要返回0;
　　只使用ANSI C/ANSI C++ 标准;
　　不要调用依赖于编译环境或操作系统的特殊函数。
　　所有依赖的函数必须明确地在源文件中 #include <xxx>
　　不能通过工程设置而省略常用头文件。


　　提交程序时，注意选择所期望的语言类型和编译器类型。

思路:

 

最简单的方法,先维护一个前缀和,然后O(n^2)枚举两个起点求出区间和判断是否%k == 0 统计个数.N <= 100000，这种方法想的满分是不可能的,但是比赛中不会做骗分也是可取的.

 

我们来想一下假设区间此时为[i,j]，那么区间和可表示为sum[j] - sum[i-1],所求为 (sum[j] - sum[i-1]  )%k == 0. 我们可以得到 当sum[j]% k == sum[i-1] % k 时 ，上式一定成立.所以我们只需要用数组记录一下 sum % k的个数分别为多少个即可,能组成的区间就是两两组合，比如num[1]=4,表示sum[j]%k==1的数量是4，那么任取一个就可以与其他三个构成区间组成k倍区间，即Cn2. 另外 % k ==0 的也要加到答案中去，表示一个单独的能被k整除的数字

#include<stdio.h>
#include<math.h>
#include<iostream>
#include<string>
#include<vector>
#include<queue>
#include<string.h>
#include<algorithm>
using namespace std;

const int maxn = 1e5+5;

int main()
{	
	long long int sum[maxn];
	long long int nu[maxn];
	int n,k;
	int ans;
	
	while(~scanf("%d%d",&n,&k)){
		long long ans = 0;
		memset(sum,0,sizeof(sum));
		memset(nu,0,sizeof(nu));
		int num;
		for(int i = 0; i < n; i++){
			cin >> num;
			sum[i+1] = sum[i]+num;  
			if (sum[i+1]%k == 0)
				ans++;
			nu[sum[i+1]%k]++;
		}
		for(int i = 0; i < k; i++){
			ans += (nu[i]*(nu[i]-1)/2);
		}
		cout << ans << endl;
	}
    return 0;
}
/*
5 2
1
2
3
4
5
*/