560. 和为 K 的子数组

题目描述

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

示例 1：

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

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

提示：

1 <= nums.length <= 2 * $10^4$
-1000 <= nums[i] <= 1000
-$10^7$ <= k <= $10^7$

C++代码

class Solution {
public:
    unordered_map<int, int> hash;
    int ans;
    
    int subarraySum(vector<int>& nums, int k) {
        int n = nums.size();
        vector<int> s(n + 1, 0);
        
        for(int i = 0; i < n; i ++)
            s[i + 1] = s[i] + nums[i];
        
        hash[0] = 1;

        for(int i = 1; i <= n; i ++){
            ans += hash[s[i] - k];
            hash[s[i]] ++; 
        }

        return ans;
    }
};