LeetCode560. Subarray Sum Equals K

一、题目

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。

子数组是数组中元素的连续非空序列。

示例 1：

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

输入：nums = [1,2,3], k = 3
输出：2

提示：

1 <= nums.length <= 2 * 104
-1000 <= nums[i] <= 1000
-107 <= k <= 107

二、题解

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int n = nums.size();
        //记录前缀和为key的子数组的数量
        unordered_map<int,int> map;
        map[0] = 1;
        int res = 0,sum = 0;
        for(int i = 0;i < n;i++){
            sum += nums[i];
            //map[sum-k]表示在i位置之前，有多少前缀和为sum-k的子数组
            res += map[sum - k];
            map[sum]++;
        }
        return res;
    }
};