560. 和为 K 的子数组

本文介绍了如何使用哈希表在O(n)时间复杂度和O(n)空间复杂度下,求解给定数组中和为目标值k的子数组数量。通过遍历数组,更新前缀和并检查哈希表中是否存在前缀和差值,统计满足条件的子数组数量。
摘要由CSDN通过智能技术生成
思路

给定数组 nums 和目标值 k,要找到子数组的数量,使其元素之和等于 k。为了解决这个问题,可以使用一个哈希表来存储之前出现的前缀和,并统计出现次数。遍历数组过程中,不断更新前缀和,并检查之前的前缀和中是否存在 sum[i] - k,如果存在,则说明找到了一个子数组的和为 k,累加相应的计数。

解题方法

初始化一个哈希表,用于存储前缀和及其出现次数。同时初始化前缀和为 0 的出现次数为 1。
初始化变量 sum 为 0,用于记录当前位置的前缀和。
初始化变量 ret 为 0,用于记录子数组的数量。
遍历数组 nums:
更新前缀和 sum += nums[i]。
检查当前位置之前是否存在前缀和为 sum - k 的情况,若存在,则累加相应的计数到 ret 中。
更新哈希表中前缀和为 sum 的出现次数。
返回 ret,即为满足条件的子数组数量

复杂度

时间复杂度: O(n)

空间复杂度: O(n)

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
    unordered_map<int, int> hash; // 统计前缀和出现的次数 
    hash[0] = 1;
    int sum = 0, ret = 0; 
    for(auto x : nums) 
    {
        sum += x; // 计算当前位置的前缀和 
        if(hash.count(sum - k)) ret += hash[sum - k]; // 统计个数 
        hash[sum]++;
    }
      return ret;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值