【leetcode刷题日记】lc.560-和为 K 的子数组

目录

1.题目

2.代码

---

1.题目

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

2.代码

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