有负数,不能用滑动窗口了。考虑前缀和思想,用一个map记录从0到i的每个和是否出现过,用让res+=ma[当前和-k],为什么是+=,而不是判断有无这个键然后+1呢,因为有负数啊!!我们需要的那个前缀和可能多次出现了,所以res+上所有从0到i出现这个前缀和的个数。
举个例子,我们有1,2,3,然后k是5,累加和最后到6,所以我们只要减去一段和为1的前缀和即可,用ma判断和为1的前缀和有几种情况,全让res加上
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
//不是滑动窗口,要考虑负数,不能收缩left
//如果是二维dp也有点麻烦,就是记录从几到几
//vector<vector<int>> dp;
//好像二维dp和普通的暴力法一样...
//好像还是有滑动窗口那个味的,滑动窗口O(N)在于left一次能走很多步
//前缀和,不是普通的前缀和,因为前缀和也要O(N^2)的时复
//相当于和为0的和已经出现过一次了,也就是一个数字都不取
unordered_map<int,int> ma{{0,1}};
int res = 0;
int sum = 0;//记录从0到i的和为多少,然后判断sum-k的前缀和有几个
//意思就是123,我们要求和为5的,就一路加,1,3,6,最后一次判断ma[6-5]不为0,也就是有一段从0到某一个数字x的区间值为1,那么从x到当前数字的和就是5
for(int x:nums){
sum += x;
res += ma[sum-k];
ma[sum]++;
}
return res;
}
};