【Leetcode】560. 和为K的子数组&974. 和可被 K 整除的子数组（前缀和+哈希表）

 

 

public class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0, pre = 0;
        HashMap < Integer, Integer > map = new HashMap < > ();
        map.put(0, 1);
        for (int i = 0; i < nums.length; i++) {
            pre += nums[i];
            if (mp.containsKey(pre - k))
                count += mp.get(pre - k);
            mp.put(pre, mp.getOrDefault(pre, 0) + 1);
        }
        return count;
    }
}

 

 

 

 

class Solution {
    public int subarraysDivByK(int[] A, int K) {
        //记录余数出现的次数
        HashMap<Integer, Integer> map = new HashMap<>();
        map.put(0,1);
        int pre = 0;//前缀和
        int ans = 0; 
        for(int i = 0; i < A.length; i++){
            pre += A[i];
            int m = (pre % K + K) % K;//余数 （负数处理）
            int same = map.getOrDefault(m,0);
            ans += same;
            map.put(m,same+1);
        }
        return ans;
    }
}