【LeetCode 560】和为K的子数组（前缀和+哈希）

最新推荐文章于 2025-05-01 18:13:28 发布

短尾黑猫

最新推荐文章于 2025-05-01 18:13:28 发布

阅读量784

点赞数 21

分类专栏： LeetCode刷题文章标签：哈希算法 leetcode 算法

本文链接：https://blog.csdn.net/qq_45371462/article/details/147618058

版权

LeetCode刷题专栏收录该内容

8 篇文章

订阅专栏

题面：

在这里插入图片描述

思路：

看到连续非空序列之和，容易想到前缀和计算差分， $n u m s$ 区间 $[j, i]$ 的和即为： $s u m (i, j) = p re [i] - p re [j - 1]$ 。
那么要怎么计算出有多少个区间符合题意 $s u m (i, j) == k$ 呢？

令 $p re [i]$ 表示 $\sum_{j=0}^i nums[j]$ ，即 $[0, ... i]$ 区间内所有数的和。

这就转化为，对于每个点 $i$ ，要求有多少个 $j,j\in[0,i]$ 是满足 $s u m (i, j) = p re [i] - p re [j - 1] = k$ ，那么就会有多少对区间 $[j, i]$ 对答案有贡献。

对于等式 $k\ (i\in[0,n),j\in[1,i])$ ，可以化为： $p re [j - 1] = p re [i] - k$ 。
这时候我们可以只统计合法的，即只统计前缀和为 $p re [i] - k$ 的 $p re [j]$ 有多少个即可。（这时候就转化为力扣 hot 第一题 两数之和了）
写个哈希表 $m p$ ，以前缀和为 $k ey$ ，出现次数为对应的 $v a l$ ，记录 $p re [i]$ 出现的次数。然后从左往右更新 $m p$ 并统计贡献，那么当前点 $i$ 的贡献 $m p [p re [i] - k]$ 可在 $O (1)$ 时间内得到。最后的答案即为：以所有下标 $i$ 为区间结尾的贡献之和。

注意： 因为已经从左往右遍历更新 $m p$ 了，所以能保证在当前点 $i$ 下统计的 $p re [j]$ 都是在 $i$ 之前的，即 $j\in[0,i]$ ，且每个前缀和 $p re [i]$ 只和上一个 $p re [i - 1]$ 有关，故拿一个临时变量记录 $p re [i - 1]$ 即可得到当前 $p re [i]$ 。

代码：

int subarraySum(vector<int>& nums, int k) {
    int cnt = 0, pre = 0;
    unordered_map<int, int> mp;
    //考虑pre[i]==k的情况，即pre[j-1]=pre[i]-k=0
    //所以可将mp[0]初始化为1，当然，也可以直接判断 pre==k?++cnt:next()
    mp[0] = 1; 
    for(const auto& x : nums) {
        pre += x;
        if(mp.find(pre - k) != mp.end())
            cnt += mp[pre - k];
        //我们是后推入mp[pre]的，否则如果先 mp[pre]++，则当 k=0 时会出错
        mp[pre]++;
    }
    return cnt;
}