蓝桥杯 k倍区间【第八届】【省赛】【B组】(Java)

问题描述
给定一个长度为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

思路：
第一时间都会想到用前缀和来处理，若暴力求解，即用i和j来表示区间，然后算i-j区间的和是否是k的倍数的话，复杂度会达到O(n平方),本题会超时，只能通过两个测试点。然后可以用同余定理，若a%k==b%k,那么|b-a| % k一定等于0，即一定是k的倍数。

我用的方法是将相同余数的点能组成的区间数加起来（因为任意两个余数相同的前缀和之差一定是k的倍数），看最终能组成多少区间。获取每个前缀和对k取模的余数，余数一定是0~k-1，然后对应余数的数量就加一，计算每个余数的数量有多少个，最后把各个余数数量转化成区间，例如，假设余数为1的数量有4个，那么他们可以形成的区间有3+2+1=（（4-1）+1）*n-1 / 2 = 6个。

特别注意：余数为0的可以自己组成区间[i,i]，所以计算余数为0的区间是，要加上本身的点数量，其余余数，若点的数量只有一个，那么不能组成区间。

完整代码：

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
		String[] s1 = reader.readLine().trim().split(" ");	//	获取n和k
		int n = Integer.parseInt(s1[0]);
		int k = Integer.parseInt(s1[1]);
		int[] sum = new int[n+1];		//	定义前缀和数组
		long[] count = new long[k];		//	存放每个余数的数量的数组，0~k-1
		long res = 0;					//	定义结果初始为0
 		for (int i = 0; i < n; i++) {	//	直接在计算前缀和的时候就计算余数，节约时间
			sum[i+1] = sum[i]%k + (Integer.parseInt(reader.readLine()))%k;	//相加后再取模等于，先取模后相加 
			sum[i+1] %= k;				//	得到的结果可能大于k，再取一次模
			count[sum[i+1]]++;			//	对于余数的数量加一
		}
 		res += count[0];				//	余数为0，自身就可以构成区间
 		for (int i = 0; i < k; i++) {
			if (count[i]>1) {			//	其余余数必须大于1才能构成区间
				res += find(count[i]);	//	若数量大于2，则加上能构成的区间数
			}
		}
 		System.out.print(res);
	}
	private static long find(long n) {	//	获取能构成区间数的方法
		return n*(n-1)/2;
	}
}

蓝桥杯AC结果：