题目链接:
560. 和为 K 的子数组 - 力扣(LeetCode)
题目描述:
给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k的子数组的个数。
子数组是数组中元素的连续非空序列。
示例 1:
输入:nums = [1,1,1], k = 2
输出:2
示例 2:
输入:nums = [1,2,3], k = 3
输出:2
提示:
1 <= nums.length <= 2 * 104-1000 <= nums[i] <= 1000-107 <= k <= 107
我的思路:
我们可以创建一个哈希表来记录每一个前缀和的出现次数,然后通过当前前缀和来减k得到数和哈希表的数来对比,如果哈希表也有这个数,说明之前有前缀和可以和当前的前缀和相减可以得到k,所以答案加上当前前缀和-k得到的哈希映射的个数,也就是numQ - numO = k中numO的个数
我的代码:
java
class Solution {
public int subarraySum(int[] nums, int k) {
HashMap<Integer, Integer> integerIntegerHashMap = new HashMap<>();
integerIntegerHashMap.put(0,1);
int ans=0;
int numQ=0;
for (int num : nums) {
numQ += num;
if (integerIntegerHashMap.containsKey(numQ - k)) {
ans += integerIntegerHashMap.get(numQ - k);
}
integerIntegerHashMap.put(numQ, integerIntegerHashMap.getOrDefault(numQ, 0) + 1);
}
return ans;
}
}