和为k的子数组

    xiaoxiao2022-06-26  155

    题目描述:

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

    样例:

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

    数组的长度为 [1, 20,000]。数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。

    分析:

    这道题和LeetCode523题:连续的子数组和类似。 定义map变量用来存储前缀和以及前缀和的出现次数, 当前缀和-k在map中存在时,证明前缀和-k和前缀和之间的元素符合要求。同一个sum可能对应不同的前缀,如果sum在map中已经存在,则更新sum对应的value在原来的次数上加1

    //执行用时 : 36 ms, 在Subarray Sum Equals K的Java提交中击败了84.91% 的用户 //内存消耗 : 49.1 MB, 在Subarray Sum Equals K的Java提交中击败了20.83% 的用户 public int subarraySum(int[] nums, int k) { Map<Integer,Integer> map=new HashMap<>(); map.put(0, 1); int sum=0; int count=0; for(int i=0;i<nums.length;i++) { sum+=nums[i]; if(map.containsKey(sum-k)) count+=map.get(sum-k); if(map.containsKey(sum)) map.put(sum, map.get(sum)+1); else map.put(sum, 1); } return count; }

    最新回复(0)