560. 和为 K 的子数组(前缀和解决子串问题)

发布时间 2023-07-24 14:29:36作者: xiazichengxi

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


示例 1:

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

> 思路

  • 每个元素对应一个“前缀和”
  • 遍历数组,根据当前“前缀和”,在 map 中寻找「与之相减 == k」的历史前缀和
  • 当前“前缀和”与历史前缀和,差分出一个子数组,该历史前缀和出现过 c 次,就表示当前项找到 c 个子数组求和等于 k。
  • 遍历过程中,c 不断加给 ans,最后返回 ans

> 代码


class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int sum = 0, ans = 0;
        unordered_map<int,int> mp;
        mp[0] = 1;
        for(int i: nums){
            //找所有的前缀和
            sum += i;
            if(mp.find(sum-k) != mp.end()) ans += mp[sum-k];
            mp[sum] ++;
        }
        return ans;
    }
};