原题链接
1658. Minimum Operations to Reduce X to Zero
题目描述
给你一个整数数组 nums 和一个整数 x 。每一次操作时,你应当移除数组 nums 最左边或最右边的元素,然后从 x 中减去该元素的值。请注意,需要 修改 数组以供接下来的操作使用。
如果可以将 x 恰好 减到 0 ,返回 最小操作数 ;否则,返回 -1 。
示例 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 。
解题思路
使用滑动窗口找中间最长的片段,使得 sum(片段) = sum(nums)-x
代码
class Solution {
public:
int minOperations(vector<int>& nums, int x) {
int min_step = INT_MAX;
int sum = accumulate(nums.begin(), nums.end(), 0);
if(x > sum) return -1;
int l = 0, r = 0, max_len = -1, cur_sum = 0, n = nums.size();
while(l < n){
if(r < n) cur_sum += nums[r++];
while(cur_sum > sum-x && l < n) cur_sum -= nums[l++];
if(cur_sum == sum-x) max_len = max(max_len, r-l);
if(r == n) l++;
}
return max_len == -1 ? -1 : n - max_len;
}
};