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;
    }
};
原文地址:https://www.cnblogs.com/xgbt/p/13463795.html