1658. Minimum Operations to Reduce X to Zero

原题链接

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;
    }
};