5471. 和为目标值的最大数目不重叠非空子数组数目 前缀和
给你一个数组 nums 和一个整数 target 。
请你返回 非空不重叠 子数组的最大数目,且每个子数组中数字和都为 target 。
示例 1:
输入:nums = [1,1,1,1,1], target = 2
输出:2
解释:总共有 2 个不重叠子数组(加粗数字表示) [1,1,1,1,1] ,它们的和为目标值 2 。
示例 2:
输入:nums = [-1,3,5,1,4,2,-9], target = 6
输出:2
解释:总共有 3 个子数组和为 6 。
([5,1], [4,2], [3,5,1,4,2,-9]) 但只有前 2 个是不重叠的。
示例 3:
输入:nums = [-2,6,6,3,5,4,1,2,8], target = 10
输出:3
示例 4:
输入:nums = [0,0,0], target = 0
输出:3
提示:
1 <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4
0 <= target <= 10^6
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-number-of-non-overlapping-subarrays-with-sum-equals-target
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
用sum记录当前的前缀和,用set存放各个位置的前缀和。
在记录的过程中,对于当前的sum,若set中存在 sum - target 的值,这就代表找到一个子数组。
为了防止重叠,即之前所有的前缀和都不用了,直接把set清空。
class Solution {
public:
int maxNonOverlapping(vector<int>& nums, int target) {
set <int> s;
int ans = 0, sum = 0;
s.insert(0);
for (auto num : nums) {
sum += num;
if (s.find(sum - target) != s.end()) {
s.clear();
sum = 0;
ans++;
}
s.insert(sum);
}
return ans;
}
};