【数据结构-哈希&&前缀】【滑动窗口】力扣930.和相同的二元子数组

给你一个二元数组 nums ,和一个整数 goal ,请你统计并返回有多少个和为 goal 的 非空 子数组。

子数组 是数组的一段连续部分。

示例 1:
输入:nums = [1,0,1,0,1], goal = 2
输出:4
解释:
有 4 个满足题目要求的子数组:[1,0,1]、[1,0,1,0]、[0,1,0,1]、[1,0,1]

示例 2:
输入:nums = [0,0,0,0,0], goal = 0
输出:15
在这里插入图片描述

哈希+前缀

class Solution {
public:
    int numSubarraysWithSum(vector<int>& nums, int goal) {
        int sum = 0;
        unordered_map<int, int> cnt;
        int ret = 0;
        cnt[0] = 1;
        for (auto& num : nums) {
            sum += num;
            ret += cnt[sum - goal];
            cnt[sum]++;
        }
        return ret;
    }
};

时间复杂度:O(n),其中 n 为给定数组的长度。对于数组中的每个元素,我们至多只需要插入到哈希表中中一次。
空间复杂度:O(n),其中 n 为给定数组的长度。哈希表中至多只存储 O(n) 个元素。

官解的代码比较抽象,对其进行了微调。在这个算法中,举个例子[1,0,1,0,1]作为我们的nums,goal = 2。cnt是一个哈希表,用来储存子数组相同的和的数量。想象有一个指针left,left = sum - goal,然后再想象有一个指针right指向当前num。
在这里插入图片描述

有以上四种情况,所以输出为4。

滑动窗口

在这里插入代码片
  • 10
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值