剑指 Offer II 010. 和为 k 的子数组

给定一个整数数组和一个整数 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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值