给定一个整数数组和一个整数 k ,请找到该数组中和为 k 的连续子数组的个数。
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int,int> imap;
int cursum = 0;
int result = 0;
imap[0] = 1;
for(int i = 0; i < nums.size(); ++ i){
cursum += nums[i];
if(imap.count(cursum - k)){
result +=imap[cursum - k];
}
imap[cursum] ++;
}
return result;
}
};
示例:
3 4 2 5 6 2 1
0 1 2 3 4 5 6
target = 7
imap[0] = 1
cursum = 3
imap[0] = 1 imap[3] = 1
cursum = 7
imap[0] = 1 imap[3] = 1
result = 1
imap[0] = 1 imap[3] = 1 imap[7] = 1
cursum = 9
imap[0] = 1 imap[3] = 1 imap[7] = 1 imap[9] = 1
cursum = 14
result = 2
imap[0] = 1 imap[3] = 1 imap[7] = 1 imap[9] = 1 imap[14] = 1
cursum = 20
result = 3
imap[0] = 1 imap[3] = 1 imap[7] = 1 imap[9] = 1 imap[14] = 1 imap[20] = 1
cursum = 22
imap[0] = 1 imap[3] = 1 imap[7] = 1 imap[9] = 1 imap[14] = 1 imap[20] = 1 imap[22] = 15
cursum = 23
imap[0] = 1 imap[3] = 1 imap[7] = 1 imap[9] = 1 imap[14] = 1 imap[20] = 1 imap[22] = 15 imap[23] = 16
明白了
我的脑子真的不如计算机
计算机真的不如天才的大脑
前缀和的变相
每走一步,计算当前位置到起始位置的和,同时unordered_map中存储的是前面每一个位置到起始位置的和
如果当前位置总和减去k的值能在unordered_map中找到,那么代表找到的那个值所代表的位置前进一步到当位置之间的和
就是题目要求的区间和
如下通用的前缀和解法,如上则是变相前缀和解法
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
int len = nums.size();
int ret = 0;
// sums[i] 为 前i个元素的和
vector<int> sums(len + 1, 0);
for(int i = 1; i <= len; i++) {
sums[i] = sums[i-1] + nums[i-1];
for(int j = 0; j < i; j++) {
// 计算前i个元素和-前j个元素和 == k ?
// 第j、j+1、...、i-1个元素的和
if(sums[i] - sums[j-1] == k)
ret++;
}
}
return ret;
}
};
3 4 2 5 6 2 1
0 1 2 3 4 5 6
target = 7
sums:
0 3 7 9 14 20 22 23
0 1 2 3 4 5 6 7