Leetcode-560. 和为K的子数组

560. 和为K的子数组
给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :

数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。
解题思路
提供了两种思路：一种是普通的记录下前缀数组和，每次求子集和时，则直接遍历前缀数组和进行遍历
优化解法是：我不仅记录下前缀和，还记下前缀和出现的次数。这样每次得到一个前缀和的时候，我就可以直接求得它需要匹配的前缀和的次数，并进行结果更新

class Solution {
    /*/常规解法：记录下前缀和数组
    public int subarraySum(int[] nums, int k) {
        int n=nums.length;
        //前缀和数组
        int[] sum=new int[n+1];
        sum[0]=0;
        for(int i=0;i<n;i++){
            sum[i+1]=sum[i]+nums[i];
        }
        int res=0;
        for(int j=1;j<=n;j++){
            for(int i=0;i<j;i++){
                if(sum[j]-sum[i]==k){
                    res++;
                }
            }
        }
        return res;
    }*/
    //优化解法：记录前缀和和对应的前缀和的出现次数
    public int subarraySum(int[] nums, int k) {
        int n=nums.length;
        //前缀和数组
        HashMap<Integer,Integer> put_sum=new HashMap<>();
        // base case
        put_sum.put(0, 1);
        int res=0;
        int sum_i=0;
        for(int i=0;i<n;i++){
            sum_i+=nums[i];
            int sum_j=sum_i-k;
            //直接查找合适的前缀和的个数
            if(put_sum.containsKey(sum_j)){
                res+=put_sum.get(sum_j);
            }
            put_sum.put(sum_i,put_sum.getOrDefault(sum_i,0)+1);

        }
        return res;
    }

}