1658. 将 x 减到 0 的最小操作数
给你一个整数数组 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
逆向思维,滑动窗口
把问题转换成「从 nums中移除一个最长的子数组,使得剩余元素的和为 x」。把离散变成连续。
正数保证了窗口和的单调性,所以可以用滑动窗口。如果窗口和等于sum-target,则找到了一个可行解。
如果存在负数的话就没有这个性质了,也就不能确定什么时候扩大和缩小窗口,也就不能使用滑动窗口算法,而应该使用 前缀和 + 哈希表的方式 解决,参见 560. 和为K的子数组。
class Solution {
public int minOperations(int[] nums, int x) {
int n = nums.length, sum = 0;
// 计算总和
for (int i = 0; i < n; i++) {
sum += nums[i];
}
// 滑动窗口需要寻找的子数组目标和
int target = sum - x;
// 滑动窗口左右边界
int left = 0, right = 0;
// 记录窗口内所有元素和
int windowSum = 0;
// 记录目标子数组的最大长度
int maxLen = Integer.MIN_VALUE;
// 开始执行滑动窗口框架
while (right < nums.length) {
// 扩大窗口
windowSum += nums[right];
right++;
while (windowSum > target && left < right) {
// 缩小窗口
windowSum -= nums[left];
left++;
}
// 寻找目标子数组
if (windowSum == target) {
maxLen = Math.max(maxLen, right - left);
}
}
// 目标子数组的最大长度可以推导出需要删除的字符数量
return maxLen == Integer.MIN_VALUE ? -1 : n - maxLen;
}
}
- 时间复杂度:O(n),其中 n 为 nums 的长度。left 和 right 均最多遍历整个数组一次。
- 空间复杂度:O(1),仅用到若干额外变量。