今日十题：560.和为k的子数组的个数

冲帕Chompa

于 2025-07-30 14:38:39 发布

阅读量259

点赞数 4

CC 4.0 BY-SA版权

文章标签：算法数据结构哈希算法

本文链接：https://blog.csdn.net/qq_59259601/article/details/149776886

解法：子数组的和考虑前缀和 + 哈希

思想：

前缀和的差值一定是连续子数组的和
可以用一个表来存已经计算过的前缀和pre1及其出现的次数
遍历数组每次计算完当前下标对应的前缀和pre2，就查看是否存在pre2 - k = pre1，更新结果
处理细节问题，可能某个前缀和本身等于k，所以首先要在表里插入0，即pre == k -> pre - k = 0

代码如下：

    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int, int> umap;
        int res = 0;
        int presum = 0;
        umap[0] = 1;
        for (int j = 0; j < nums.size(); j++)
        {
            presum += nums[j];
            if (umap.count(presum - k)) res += umap[presum - k];
            umap[presum]++; 
        }
        return res;
    }