刷题-力扣-1004. 最大连续1的个数 III

1004. 最大连续1的个数 III

题目链接

来源：力扣（LeetCode）
链接：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 <= A.length <= 20000
• 0 <= K <= A.length
• A[i] 为 0 或 1

题目分析

1. 根据题目描述选取替换K个0后最长连续的1的长度
2. 构建一个滑动窗口，保证窗口内最多包含K个0
3. left和right就是窗口的左边和右边
4. 当right的下一位为1时，将right的下一位加入到窗口中，即right++
5. 当right的下一位为0时，让窗口的左侧出一个0，即当left++，直到A[left]==0为止
6. 把窗口的最大长度保存在len中，len即为所求

代码

class Solution {
public:
    int longestOnes(vector<int>& A, int K) {
        int len = 0;
        int left = 0;
        int right = 0;
        while (left < A.size() && right < A.size()) {
            if (A[right] == 1 || K > 0) {
                if (A[right] == 0) K--;
                right++;
                
            } else {
                if (A[left] == 0) right++;
                left++;
            }
            len = len > right - left ? len : right - left;
        }
        return len;
    }
};