给你一个整数数组 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
哎,如何找到map的键值对分别对应着谁是关键,pre是key,pre次数是value,当然了,暴力的话挺简单的,两遍for循环,用前缀和pre-k正好在hash表里面找到,说明当前数的前面包含符合要求的前缀和,而且pre-k对应的键就是要找的这个数的前面的前缀和里符合要求的次数。晕了,做10次都不一定有一眼看穿的感觉,太难了。
class Solution {
public int subarraySum(int[] nums, int k) {
int count=0,pre=0;
Map<Integer,Integer> map=new HashMap<Integer,Integer>();
map.put(0,1);
for(int x:nums){
pre+=x;
if(map.containsKey(pre-k)){
count+=map.get(pre-k);
}
map.put(pre,map.getOrDefault(pre,0)+1);
}
return count;
}
}