给你一个整数数组 nums
和一个整数 x
。每一次操作时,你应当移除数组 nums
最左边或最右边的元素,然后从 x
中减去该元素的值。请注意,需要 修改 数组以供接下来的操作使用。
如果可以将 x
恰好 减到 0
,返回 最小操作数 ;否则,返回 -1
。
示例 1:
输入:nums = [1,1,4,2,3], x = 5 输出:2 解释:最佳解决方案是移除后两个元素,将 x 减到 0 。
示例 2:
输入:nums = [5,6,7,8,9], x = 4 输出:-1
示例 3:
输入:nums = [3,2,20,1,1,3], x = 10 输出:5 解释:最佳解决方案是移除后三个元素和前两个元素(总共 5 次操作),将 x 减到 0 。
提示:
1 <= nums.length <= 10^5
1 <= nums[i] <= 10^4
1 <= x <= 10^9
提示 1
Think in reverse; instead of finding the minimum prefix + suffix, find the maximum subarray.
提示 2
Finding the maximum subarray is standard and can be done greedily.
解法:滑动窗口
class Solution {
public int minOperations(int[] nums, int x) {
int n = nums.length;
int ans = Integer.MAX_VALUE;
// nums中所有数的和
long sum = 0;
// 窗口内数的和
long window = 0;
int left = 0;
int right = 0;
for (int i = 0; i < n; i++) {
sum += nums[i];
}
while (right < n) {
window += nums[right];
while (sum - window < x && left <= right) {
window -= nums[left];
left++;
}
if (sum - window == x) {
ans = Math.min(ans, n - (right - left + 1));
}
right++;
}
return ans == Integer.MAX_VALUE ? -1 : ans;
}
}
复杂度分析
- 时间复杂度:O(n),n 是 数组的长度。
- 空间复杂度:O(1)。