蓝桥杯 2017-10 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

看到倍数，首先想到的是倍数dp问题，但是还是和背包一样不太熟悉，独立写代码写到一半就卡了，然后看别人的思路，只能说，学到了。
这个题两个关键字“倍数”“连续的数的和”所以这个题要结合取余和前缀和，（但是我之前的两种思路都是只考虑了其中的一个，我之前写的代码就是考虑的前缀和，但只考虑了前缀和，所以时间超限，然后就是现在想到的倍数dp没写出来）。
反思一下，我之前用前缀和时间主要花在枚举两个前缀和的差是否是k的倍数上，但实际结合一个函数保存余数的话，不需要无脑枚举，余数相等的两个前缀和相减一定是k的倍数。所以k的倍数的和的个数分两部分，一部分就是前缀和，另一部分是前缀和的差。
第一部分好求，直接求前缀和判断就行，第二部分，有一点小小的弯。首先我自己想到的一个方法是在算出所有的和的余数后（sum[i]:表示余数是i的和的个数），可以形成的差的个数是n（n - 1）/ 2.但是在看别人代码时不得不说“妙啊”。他是在求前缀和的过程中求出了差的数量，也就是在求前缀和的过程中计算当前和与前面的差是k的倍数的个数（该个数等于sum[i] - 1（计算i与前面的差是k的倍数的个数），所以可以先求差，再对sum数组加1）.这部分代码

for (int i = 1; i < n; i++) {
			sum[i] = (sum[i - 1] + a[i]) % k;
			res += dp[sum[i]];
			dp[sum[i]]++;
		}

然而一开始没有AC，我也没有错误测试样例，第一天实在找不到了，就第二天再看。几乎是经常犯得错误了。就是下标从0开始和从1开始的问题。在这一题，下标从一开始比较好写，因为他有i-1.但是下标从0开始就需要多考虑点东西，还容易错。
第一错：

for (int i = 0; i < n; i++) {
		sum[i] = (sum[i - 1] + a[i]) % k;

i从0开始但是下面i - 1，就是-1了，但奇怪的是我编译测试样例竟然通过了。
第二错

sum[0] = a[0] % k;		
		for (int i = 1; i < n; i++) {
			sum[i] = (sum[i - 1] + a[i]) % k;
			res += dp[sum[i]];
			dp[sum[i]]++;
		}

因为下标从一开始了，我要把下标为0单独拿出来写，如上面所示，但是啊但是。我忘了dp[sum[i]]++;，因为是第一项，所以没有差，不用考虑res += dp[sum[i]];
所有代码

import java.util.Scanner;

public class okt10k倍区间 {
	public static void main(String[] args) {

		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		int k = scanner.nextInt();
		int[] a = new int[n];
		int[] sum = new int[n];
		int[] dp = new int[n];
		for (int i = 0; i < n; i++) {
			a[i] = scanner.nextInt();
		}
		sum[0] = a[0] % k;
		long res = 0;
		dp[sum[0]]++;
		// res += dp[sum[0]];

		for (int i = 1; i < n; i++) {
			sum[i] = (sum[i - 1] + a[i]) % k;
			res += dp[sum[i]];
			dp[sum[i]]++;
		}
		System.out.println(res + dp[0]);

	}
}