算法系列——和为 K 的子数组(Subarray Sum Equals K)

Posted 2022-07-16 BridgeGeorge

题目

给你一个整数数组 nums 和一个整数 k ，请你统计并返回该数组中和为 k 的连续子数组的个数。

示例 1：

输入：nums = [1,1,1], k = 2
输出：2
示例 2：

输入：nums = [1,2,3], k = 3
输出：2

解答

哈希表+前缀和 时间复杂度为O(n)，空间复杂度O(n)

class Solution 
    public int subarraySum(int[] nums, int k) 
        //前缀和 
        // pre[i] [0,i]的和
        //[i,j] pre[j]-pre[i-1]=k
        int sum=0;
        int count=0;
        Map<Integer,Integer> map=new HashMap<>();
        map.put(0,1);//此处必须赋值为1
        for(int i=0;i<nums.length;i++)
            sum+=nums[i];
            if(map.containsKey(sum-k))
                count+=map.get(sum-k);
            
            map.put(sum,map.getOrDefault(sum,0)+1);   
        
        return count;