滑动窗口遍历连续子数组_LeetCode1658

题目描述

给你一个整数数组 nums 和一个整数 x 。每一次操作时，你应当移除数组 nums 最左边或最右边的元素，然后从 x 中减去该元素的值。请注意，需要 修改 数组以供接下来的操作使用。

如果可以将 x 恰好 减到 0 ，返回 最小操作数 ；否则，返回 -1 。

提示：

• 1 <= nums.length <= 10^5
• 1 <= nums[i] <= 10^4
• 1 <= x <= 10^9

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimum-operations-to-reduce-x-to-zero

解题思路

题意转换：删除数组两端的元素使删除元素和为$x$，等价于找到原数组中间某段连续子数组且和为$sum-x$。

遍历子数组的方法：滑动窗口，设左下标$left$、右下标$right$、数组当前和$cur$

• 当$cur=sum-x$：找到满足要求子数组，更新结果，$right$右移

• 当$cur<sum-x$：$right$右移使$cur$增大

• 当$cur>sum-x$：$left$左移使$cur$减小

不断循环，直到$right$超出数组范围即可。

时间复杂度：由于$left$和$right$最多都是遍历$N$次（$N$为数组长度），因此循环次数最多为$2N$次，时间复杂度度$O(N)$。

代码

class Solution {
public:
    int minOperations(vector<int>& nums, int x) {
        int N = nums.size(), sum = 0;
        for(int i = 0; i < N; i++) sum += nums[i];
        int target = sum - x; //目标：找到子数组和为target
        if(target==0) return N;   //全部减掉正好减到0
        else if(target < 0) return -1;  //sum < x 无法满足要求
        //left 和 right 为遍历子数组的滑动窗口指针
        int ans = -1, left = 0, right = left + 1, cur = nums[left]; 
        while(left <= right && right <= N){
            //找到子数组， 更新ans
            if(cur == target){
                ans = max(ans, right - left);
            }
            if(cur <= target){
                cur += nums[right++];
            }
            else{
                cur -= nums[left++];
            }
        }
        return ans == -1 ? ans:(N-ans);       
    }
};