给你一个二元数组 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。
滑动窗口
在这里插入代码片