题目
给你一个数组 nums 和一个整数 target 。
请你返回 非空不重叠 子数组的最大数目,且每个子数组中数字和都为 target 。
示例 :
输入: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
思路
从左到右遍历数组,一旦发现有某个以当前下标 i 为右端点的子数组和为 target,就给计数器ans的值加 1,并从数组nums的下标 i + 1 开始,进行下一次寻找。
那么,该如何实现呢?
从左往右遍历数组nums,并保存所有出现过的前缀和,这里使用set保存
当遍历到数组元素nums[i]时,如果set中存在元素nums[i] - target,说明找到1个子数组和为target,则计数器+1
将set清空,继续从i + 1处开始遍历数组,重复上述步骤,直到数组末尾
public int maxNonOverlapping(int[] nums, int target) {
Set<Integer> set = new HashSet<Integer>();
set.add(0);
int cnt = 0;
int sum = 0;
for (int num : nums) {
sum += num;
if (set.contains(sum - target)) {
cnt++;
set.clear();
}
set.add(sum);
}
return cnt;
}