题目描述:
给定一个整数数组和一个整数 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;
}