LeetCode 560.Subarray Sum Equals K【Java】

题目描述

560. 和为K的子数组

AC代码

/*
解题思路：前缀和+hash

定义sum[i]表示从开始加到第i位置的和，在数组[1,2,3,4]上我们就有：

    sum[0] = nums[0]
    sum[1] = nums[0] + nums[1] = sum[0] + nums[1]
    sum[2] = nums[0] + nums[1] + nums[2] = sum[0] + nums[1] + nums[2] = sum[1] + nums[2]
    sum[3] = nums[0] + nums[1] + nums[2] +nums[3] = sum[0] + nums[1] + nums[2] + nums[3] = ... ...

如果我们相求数组下标[L,R]之间的连续数组的和，比如求[1,2]之间连续数组的和，可以通过sum[2]-sum[0]
求出，那么总结一下就是sum[R]-sum[L-1]=[L,R]下标之间连续数组的和,如果sum[R]-sum[L-1]=k,
那么就是我们想要的结果了。在这里，将sum[R]-sum[L-1]=k变形为sum[L-1]=sum[R]-k，通过hash表主要
来记录sum[L-1]的情况，hash.put(key,value)代表有value个前缀和为key。


    sum[2] - sum[0] = nums[1] + nums[2]

    sum[2] - sum[1] = nums[2]

PS :如果数组下标[0,2]的连续和为k，那么有sum[2]-sum[-1] = k,所以需要提前处理sum[-1] =>map.put(0,1).
*/
class Solution {
    HashMap<Integer,Integer> hash=new HashMap<>();
    public int subarraySum(int[] nums, int k) {
        int ans=0;
        //有一个前缀和为0,其实代表的是sum[-1]=0
        hash.put(0,1);
        for(int i=0,sum=0;i<nums.length;i++){
            sum+=nums[i];
            //是否存在前缀和为sum-k的连续和
            if(hash.containsKey(sum-k)) 
                ans+=hash.get(sum-k);
            hash.put(sum,hash.getOrDefault(sum,0)+1);
        }
        return ans;
    }
}