思路:
[j,i]
这个子数组和为 k 这个条件可以转化为sum[i]−sum[j−1]==k
使用hash表存储子串和
key:0到当前位置的子串和
value:该和出现的次数
(比如[-1,1,0],和为0出现了2次)
每遍历到nums[i]
,如果sum[i]-k
的结果出现在hash
里,表示存在[j,i]
使得和为k
class Solution:
def subarraySum(self, nums: List[int], k: int) -> int:
hash={}
count=0
sum=0
for i in range(len(nums)):
sum+=nums[i]
if sum==k:#[0,i]和为k
count+=1
if (sum-k) in hash:#[i,j]和为k,i可能有多个
count+=hash[sum-k]
hash[sum]=hash.get(sum,0)+1
return count
class Solution {
public int subarraySum(int[] nums, int k) {
Map<Integer,Integer>map=new HashMap<>();
int res=0;
int sum=0;
for(int num:nums){
sum+=num;
if(sum==k){
res+=1;
}
if(map.containsKey(sum-k)){
res+=map.get(sum-k);
}
map.put(sum,map.getOrDefault(sum,0)+1);
}
return res;
}
}