1658. 将 x 减到 0 的最小操作数

penguin_bark2024-09-29 15:41

文章目录

1.题目
2.思路
3.代码

1.题目

给你一个整数数组 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 。

2.思路

正难则反，通过移除一些元素，使得剩下的元素的和等于 sum(nums) - x，其中 sum(nums) 是数组所有元素的和，那么问题就转化为：找到这个最长子数组，剩余部分的元素就是要移除的，移除的操作数就是 nums.size() - 最长子数组的长度

3.代码

c++ 复制代码

class Solution {
public:
    int minOperations(vector<int>& nums, int x) {
        int sum = 0;
        // 计算数组所有元素的总和
        for(int i = 0; i < nums.size(); ++i){
            sum+=nums[i];
        }
        int target = sum - x;
        // 如果 target 小于 0，说明无法通过操作达到目标
        if (target < 0) return -1;
        int left = 0, right = 0, len = -1;
        int cur_sum = 0;
        // 遍历数组，右边界逐步向右扩展窗口
        while(right<nums.size()){
            cur_sum+=nums[right];
            ++right;
            // 如果当前窗口的和大于 target，缩小窗口
            while(cur_sum > target){
                cur_sum-=nums[left];
                ++left;
            }
            // 如果当前窗口的和等于 target，更新最长子数组的长度
            if(cur_sum == target)
                len = max(len, right-left);
        }
        return (len == -1) ? -1 : nums.size() - len;
    }
};