C++速通LeetCode中等第7题-和为K的子数组（巧用前缀和）

程序猿练习生

已于 2024-09-19 10:35:09 修改

阅读量287

点赞数 3

分类专栏： LeetCode 文章标签：算法 leetcode 数据结构

于 2024-09-19 10:31:44 首次发布

本文链接：https://blog.csdn.net/weixin_47768406/article/details/142353493

版权

LeetCode 专栏收录该内容

53 篇文章 0 订阅

订阅专栏

巧用哈希表与前缀和，前缀和差为k的两个序号之间的数组就是满足条件的子数组，用哈希表来存放每个序号的前缀和。

前缀和就是头元素到当前序号子数组元素的和

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int, int> mp;
        mp[0] = 1;
        int count = 0, pre = 0;
        for (auto& x:nums) {
            pre += x;
            if (mp.count(pre - k)) {//看有没有前缀和差为目标k的两个序号
                count += mp[pre - k];//有的话答案加上满足的前序数量
            }
            mp[pre]++;//记录每一个序号的前缀和，前缀和相等的话，map->second计数会加一
        }
        return count;
    }
};