蓝桥杯历届试题 ：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倍区间的数目。

思路：

首先因为这道题给的最大数据范围是100000，所以用二重循环肯定会超时，所以应该尽可能找一个能降低时间复杂度的办法。
要求一段序列的和是k的倍数，可以总结的是这段距离内的和余上个k等于0，可以得出(preSum[j] - preSum[i])%k == 0推出
preSum[j]%k == preSum[i]%k,所以只需对i个数的前缀和余上k进行统计，就可以将时间复杂度降到O(n);

import java.util.*;

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt(), k = sc.nextInt();
        int[] presum = new int[n + 1];//表示n个数的前缀和
        int[] cnt = new int[n];//用来统计前缀和余k的值相等的数
        long res = 0;//存储最后的结果
        //这里因为当一个前缀和余k等于0的时候，这个前缀和本身就可以组成一个区间，所以还要加上cnt[0]
        for (int i = 1; i <= n; i++) {
            presum[i] = (presum[i - 1] + sc.nextInt()) % k;
            //这里先让res加上cnt，是因为当cnt的值为1的时候，不能组成一个片段，至少要两个
            //容易推出：2个数是一个区间，3个数是1+2=3个区间.....
            res += cnt[presum[i] % k];
            cnt[presum[i] % k]++;
        }
        System.out.println(res + cnt[0]);
    }
}