力扣(LeetCode)1658. 将 x 减到 0 的最小操作数(C++/Python)

题目描述

逆向思维+滑动窗口

题目分析 ： 从数组左侧和右侧，取出左侧的连续数字，右侧的连续数字，使得这些数字之和等于 x，维护最小取数次数，作为答案 。

设整个数组之和 total ，除去左侧和右侧的连续数字，剩下的数组之和等于total-x ，而且剩下的数组正好是中间的连续数字。问题转化为，求数组之和等于total-x的最长连续数组。中间剩下的数组越长，左侧和右侧使用的数组就越短。

c++

class Solution {
public:
    int minOperations(vector<int>& nums, int x) {
        int target = accumulate(nums.begin(),nums.end(),0) - x;
        if(target<0) return -1;
        int ans = -1, l = 0, r = 0 , sum = 0;
        while(r<nums.size()){
            sum += nums[r++];
            while(sum>target) sum -= nums[l++];
            if(sum==target) ans = max(r-l,ans);
        }
        return ans<0?-1:nums.size()-ans;
    }
};

python3

class Solution:
    def minOperations(self, nums: List[int], x: int) -> int:
        target = sum(nums) - x
        if target < 0:
            return -1
        ans = -1
        l = r = s = 0
        while r < len(nums):
            s += nums[r]
            r += 1
            while s > target:
                s -= nums[l]
                l += 1
            if s == target:
                ans = max(ans, r - l)
        return ans if ans < 0 else len(nums) - ans

1. 时间复杂度 : $O(n)$ ， $n$ 是数组长度 ，计算target ，维护滑动窗口的时间复杂度 $O(n)$ 。
2. 空间复杂度 : $O(1)$ ， 只使用常量级空间 。

AC

致语

• 理解思路很重要
• 读者有问题请留言，清墨看到就会回复的。