【leetcode】1004 最大连续1的个数III（滑动窗口，双指针）

题目链接：https://leetcode-cn.com/problems/max-consecutive-ones-iii/

题目描述

给定一个由若干 0 和 1 组成的数组 A，我们最多可以将 K 个值从 0 变成 1 。

返回仅包含 1 的最长（连续）子数组的长度。

示例 1：

输入：A = [1,1,1,0,0,0,1,1,1,1,0], K = 2
输出：6
解释：
[1,1,1,0,0,1,1,1,1,1,1]
粗体数字从 0 翻转到 1，最长的子数组长度为 6。
示例 2：

输入：A = [0,0,1,1,0,0,1,1,1,0,1,1,0,0,0,1,1,1,1], K = 3
输出：10
解释：
[0,0,1,1,1,1,1,1,1,1,1,1,0,0,0,1,1,1,1]
粗体数字从 0 翻转到 1，最长的子数组长度为 10。

提示：

1. 1 <= A.length <= 20000
2. 0 <= K <= A.length
3. A[i] 为 0 或 1

思路

// 滑动窗口
// 时间复杂度：O(n) 空间复杂度：O(1)
class Solution {
public:
    int longestOnes(vector<int>& A, int K) {
        int l = 0, r = 0, ret = 0;
        int zeros = 0;
        while(r < A.size()){
            if(A[r] == 0)
                zeros++;
            // 当窗口内0的个数大于k时窗口缩小
            while(zeros > K){
                if(A[l++] == 0)
                    zeros--;
            }
            ret = max(ret, r-l+1);
            r++;
        }
        return ret;
    }
};