Leetcode 560. 和为 K 的子数组

560. 和为 K 的子数组 - 力扣（LeetCode）

class Solution {
    /**
    思路：用一个map记录前缀和，key是数值合m，value是这个m出现了n次。
    这样遍历数组，比如i从0开始，遍历到i=j时，算下下标从0-j所有元素和是多少，比如是m，
    然后算下m-k这个数值有没有出现过。
    如果出现过c次，那就是之前有c种子串和是m-k，那就对应有c种子串是k的情况；
    如果没有出现过，那就是把数值和是m的放进map，用于下次计算
     */
    public int subarraySum(int[] nums, int k) {
        Map<Integer,Integer> prefixSum=new HashMap<>();
        int res=0;
        int cnt=0;
        prefixSum.put(0,1);
        for(int i=0;i<nums.length;i++){
            cnt+=nums[i];
            // map的判断key是否存在使用containsKey，这个好好记忆一下
            if(prefixSum.containsKey(cnt-k)){
                res+=prefixSum.get(cnt-k);
            }
            if(prefixSum.get(cnt)==null){
                prefixSum.put(cnt,1);
            } else {
                prefixSum.put(cnt,prefixSum.get(cnt)+1);
            }
        }
        return res;
    }
}