原題連結

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