蓝桥杯JAVA版答案——历年真题——K倍区间

更多JAVA版答案移步我的博客：蓝桥杯JAVA版答案汇总

本题考查

前缀和、同余定理

思路

前置知识：

1. 一个数对k的余数仅有k种可能，从0到k-1
2. 前缀和代表自该元素向前所有下标小于等于该元素数组元素之和
3. 假设A到B的区间前缀和对k取余的结果等于A到C的区间前缀和对k取余的结果，则这两个区间前缀和的差的绝对值也就是B到C的区间前缀和一定整除k

代码逻辑：
arr数组第i个元素是（从第0个元素开始到第i的元素）的前缀和对k的余数
record数组记录区间前缀和对k的余数等于下标的区间数，举个例子record[4]代表对k的余数为4的区间个数

1. 计算前缀和对k的余数，存储在数组arr当中，并根据余数的值，更新数组record中元素的值
2. 对每个record中的值，根据组合原理C(n,2)即(n-1)*n/2计算该余数能组成的满足整出条件的区间数
3. 最后将区间数加上record[0]，因为当该区间前缀和对k取余等于0时，代表该区间满足整除条件，所以最后总数加上record[0]

注意：最后record[i]*(record[i]-1)时数值可能超过Integer的最大值，所以使用long类型，记录满足条件的区间总数的变量result也要使用long类型

AC代码

import java.util.Scanner;
public class Main {

	public static void main(String[] args) {
		Scanner scaner = new Scanner(System.in);
		int n = scaner.nextInt();
		int k = scaner.nextInt();
		int[] arr = new int[n];
		int[] record = new int[k];
		for (int i = 0; i < n; i++) {	
			if(i==0)	arr[i] = scaner.nextInt()%k;
			else		arr[i] = (arr[i-1] + scaner.nextInt())%k;
			record[arr[i]]++;
		}
		scaner.close();	
		long result = 0;
		for (int i = 0; i < k; i++)
			result += (long)record[i]*(record[i]-1)/2;
		result += record[0];
		System.out.println(result);
	}
}