leetcode 560. 和为 K 的子数组（前缀和+哈希表）

rvlt1

于 2023-03-22 15:45:10 发布

阅读量72

点赞数

文章标签： leetcode 散列表算法

本文链接：https://blog.youkuaiyun.com/skykone1/article/details/129711459

版权

该文章介绍了一种使用前缀和与哈希表的数据结构来优化算法，解决在给定数组中查找连续子数组和等于特定值的个数问题。通过维护前缀和数组并借助哈希表存储之前出现的和，可以将时间复杂度降低到O(n)，避免了暴力求解导致的超时问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
public:
    //前缀和+哈希表
    //如果不用哈希表复杂度达到O(n2)超时
    //sums[i]表示从0到i-1的前缀和, sums[0] = 0, 因此j到i的连续和等于sums[i + 1] - sum[j]
    //求从j到i的连续和等于k的个数（j小于i，则sums[i + 1]一定大于sums[j]） 
    //<==> 求sums[i + 1] - sums[j] == k 的个数 
    //<==> 求 sums[j] == sum[i + 1] - k 的个数
    int subarraySum(vector<int>& nums, int k) {
        int sz = nums.size();
        vector<int> sums(sz + 1);
        map<int, int>mp;
        int ans = 0;
        sums[0] = 0; 
        mp[0] = 1;
        for(int i = 0; i < sz; i++){
            sums[i + 1] = sums[i] + nums[i];
            if(mp.find(sums[i + 1] - k) != mp.end()){
                ans += mp[sums[i + 1] - k];
            }
            mp[sums[i + 1]]++;
        }
        return ans;
    }
};