1230. K倍区间 Java题解 （暴力，前缀和，哈希）【第八届蓝桥杯省赛C++B组,第八届蓝桥杯省赛JAVA B组】

输入样例：

5 2
1
2
3
4
5

输出样例：

6

解题思路：

暴力：通过控制序列的左右端点。并遍历求出序列间的和，对符合条件的计数。程序最终复杂度为O(n^3)。

前缀和：仍然是控制序列的左右端点，但是在求序列和时，运用前缀和数组求解，求和复杂度为O(1)，程序最终复杂度为O(n^2)。

前缀和+哈希：

在求序列和时，用的求前缀和方式时：sum = s[r] - s[l - 1], 并验证sum%k是否为0，这里实际上是以l∈[1, r] 到 r 的序列中找出可以被k整除的数的个数。

通过观察 (s[r] - s[l - 1])%k ,其中l∈[1, r]，则 l- 1∈[0, r - 1]，即找满足 (s[r] - s[l])%k ＝0 的个数，又等价于 找使得 s[r]%k == s[l]%k 成立的个数。

当右端点为r时，可以将从左端点到r之间的取余k的情况存入cnt[]数组，即 cnt[i] = j：表示余数为i的个数有j个。从而就可以把找s[l]%k == s[r] 的个数转化为直接在哈希表中查找满足 cnt[s[r]%k]。查找的时间复杂度为O(1)，达到空间换时间。程序最终复杂度为O(n)。

Java代码：（暴力）

import java.io.*;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] split = br.readLine().split(" ");
		int n = Integer.parseInt(split[0]);
		int k = Integer.parseInt(split[1]);
		int []arr = new int[n + 1];
		for(int i = 1; i <= n; i++) {
			arr[i] = Integer.parseInt(br.readLine());
		}
		
		long ans = 0;
		for(int r = 1; r <= n; r++) {//右端点从1到n
			for(int l = 1; l <= r; l++) {//左端点从1到r
				long sum = 0;
				for(int i = l; i <= r; i++) {//求l到r之间的和
					sum += arr[i];
				}
				if(sum % k == 0) ans++;//验证是否满足条件
			}
		}
		System.out.println(ans);
	}
}

Java代码：（前缀和）

import java.io.*;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] split = br.readLine().split(" ");
		int n = Integer.parseInt(split[0]);
		int k = Integer.parseInt(split[1]);
		
		long []s = new long[n + 1];//直接求前缀和数组
		for(int i = 1; i <= n; i++) {
			s[i] += s[i - 1] + Integer.parseInt(br.readLine());
		}
		
		long ans = 0;
		for(int r = 1; r <= n; r++) {
			for(int l = 1; l <= r; l++) {
				long sum = 0;
				sum = s[r] - s[l - 1];//通过前缀和数组直接求和
				if(sum % k == 0)ans++;
			}
		}
		System.out.println(ans);
	}
}

Java代码：（前缀和+哈希）

import java.io.*;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] split = br.readLine().split(" ");
		int n = Integer.parseInt(split[0]);
		int k = Integer.parseInt(split[1]);
		long []s = new long[n + 1];
		for(int i = 1; i <= n; i++) {//前缀和数组
			s[i] += s[i - 1] + Integer.parseInt(br.readLine());
		}
		long []cnt = new long[100000];//模拟哈希表，cnt[i] = j:表示余数为i的个数有j个
		
		long ans = 0;
		cnt[0] = 1;//先提前 将0%k为0 这一个数存入
		for(int r = 1; r <= n; r++) {
			ans += cnt[(int) (s[r] % k)];//在哈希表cnt中找出与s[r]%k余数相同的个数（从0到r - 1）
			cnt[(int) (s[r] % k)]++;//将将余数为s[r]%k的个数加一
		}
		System.out.println(ans);
	}
}