LeetCode P974-- 和可被K整除的子数组 前缀和、HashMap

题目来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sums-divisible-by-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

给定一个整数数组 A，返回其中元素之和可被 K 整除的（连续、非空）子数组的数目。

示例：

输入：A = [4,5,0,-2,-3,1], K = 5
输出：7
解释： 有 7 个子数组满足其元素之和可被 K = 5 整除：
[4, 5, 0, -2, -3, 1], [5], [5, 0], [5, 0, -2, -3], [0], [0, -2, -3], [-2,-3]

要找子区间被K整除，那就要求出子区间的和，再对K取模，看是否为0。
子区间 [i , j] 的和为 preSum[ j ] - preSum[ i ] ,其中preSum为对应的前缀和。
那么根据这个规律，我们可以写出以下代码：

 public int subarraysDivByK(int[] A, int K) {
        int rs = 0;
        int len = A.length;
        int[] preSum = new int[len + 1];
        preSum[0] = 0;
        //计算前缀和
        for (int i = 0; i < len; i++) {
            preSum[i + 1] = preSum[i] + A[i];
        }
        //二次遍历前缀和，寻找子区间mod K 为0的
        for (int i = 0; i < len; i++) {
            for (int j = i; j < len; j++) {
                if ((preSum[j + 1] - preSum[i]) % K == 0){
                    rs++;
                }
            }
        }
        return rs;
    }

但是很明显，，有了二重循环，很容易就超时了，这题这个解法就超时了。那么我们怎么去进行优化呢，可以利用Hash表去维护，每次存进去的key是前缀和模K后的值，value是数量，进行一次遍历：

 public int subarraysDivByK2(int[] A, int K) {
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0,1);
        int sum = 0;
        int rs = 0;
        for (int a : A) {
            sum += a;
            // 注意 Java 取模的特殊性，当被除数为负数时取模结果为负数，需要纠正
            int modulus = (sum % K + K) % K;
            //Map集合中有这个key时，就使用这个key值，如果没有就使用默认值defaultValue，此处默认值为0
            int same = map.getOrDefault(modulus, 0);
            //直接加是因为如果两个前缀和 模 K 都相等的话，那么他们相减就会模K为0，即符合题意
            rs += same;
            map.put(modulus, same + 1);
        }
        return rs;
    }