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

Posted BridgeGeorge

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了算法系列——和为 K 的子数组(Subarray Sum Equals K)相关的知识,希望对你有一定的参考价值。

题目

给你一个整数数组 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;
    

以上是关于算法系列——和为 K 的子数组(Subarray Sum Equals K)的主要内容,如果未能解决你的问题,请参考以下文章

leetcode [560. 和为K的子数组]

leetcode [560. 和为K的子数组]

题目地址(560. 和为 K 的子数组)

Leetcode523. 连续的子数组和

[LeetCode] 560. 和为K的子数组

⭐算法入门⭐《哈希表》中等02 —— LeetCode 560. 和为K的子数组