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

链接

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

题目

给定一个整数数组和一个整数 k ，请找到该数组中和为 k 的连续子数组的个数。

示例

示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2
解释: 此题 [1,1] 与 [1,1] 为两种不同的情况

示例 2 :
输入:nums = [1,2,3], k = 3
输出: 2

说明

• 1 <= nums.length <= 2 * 10e4
• -1000 <= nums[i] <= 1000

• -10e7 <= k <= 10e7

思路

对于这种问题，我们很容易想到滑动窗口的解法，然而，这题的数组元素可以是负数，当窗口里的数字和sum大于或者小于k时，不好判断窗口该怎么滑动，因此我们采用前缀和方法。

新建一个哈希表，遍历数组每个元素，维护当前扫描到的所有数字的和，在每个位置上都将当前的sum保存进哈希表当中，为什么这样做呢？因为比如前三个元素的sum为3，而前五个元素的sum和为10，k为7，那么我们通过第五个元素的前缀和减去第三个元素的前缀和，10-3=7，那么久说明第4个元素和第5个元素组成的连续子数组符合题意。通过以上例子，我们就可以通过判断当前的sum-k在哈希表中的值（频率）来判断前面有多少个子数组符合条件。

C++ Code

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int,int> map;
        map[0]=1;
        int sum=0;
        int count=0;
        for(int i=0;i<nums.size();i++)
        {
            sum+=nums[i];
            if(map.find(sum-k)!=map.end())count+=map[sum-k];
            map[sum]++;
        }
        return count;

    }
};

这里map[0]=1的原因是：针对例1这种情况，从下标为0开始的元素就存在符合题意的子数组。