560. Subarray Sum Equals K

Given an array of integers and an integer k, you need to find the total number of continuous subarrays whose sum equals to k.

Example 1:

Input:nums = [1,1,1], k = 2
Output: 2

Note:

  1. The length of the array is in range [1, 20,000].
  2. The range of numbers in the array is [-1000, 1000] and the range of the integer k is [-1e7, 1e7].

思路:暴力求解存在大量的重复计算,比如:在下标为0的时候,运算过 nums[0] + nums[1] + nums[2],在下标为1的时候,又算了一次nums[1] + nums[2]。

          但是,采用累加和的方式:nums[1] + nums[2] = sum[2] - sum[0],只需要遍历一次就可以知道所有的前缀和,存在map里,用的时候就可以实现在常数时间的查找。

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int cur = 0, res = 0;
        unordered_map<int, int> um;
      	// 注意这里前缀和多了一个0,防止漏掉数组的前缀和刚好等于k的情况
        um[0] = 1;
        for (int num : nums) {
            cur += num;
            res += um.find(cur - k) == um.end() ? 0 : um[cur - k];
            ++um[cur];
        }
        return res;
    }
};

只需要遍历一次就可以知道所有的前缀和,存在map里,用的时候就可以实现在常数时间的查找。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值