【leetcode】1004. 最大连续1的个数 III（max-consecutive-ones-iii）（滑动窗口）[中等]

链接

https://leetcode-cn.com/problems/max-consecutive-ones-iii/

耗时

解题：21 min
题解：8 min

题意

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

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

提示：

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

思路

滑动窗口中包含的是仅包含 1 或者包含 1 以及小于 K 个 0 的（连续）子数组。滑动窗口的最大窗口长度即是答案。

时间复杂度：$O(n)$

AC代码

class Solution {
public:
    int longestOnes(vector<int>& A, int K) {
        int n = A.size();
        int max_1 = 0;
        int rest_K = K;
        int l = 0, r = 0;
        while(r < n) {
            if(A[r] == 1) {
                r++;
            }
            else if(A[r] == 0 && rest_K > 0) {
                rest_K--;
                r++;
            }
            else {
                if(A[l] == 0) {
                    rest_K++;
                }
                max_1 = max(max_1, r-l);
                l++;
            }
        }
        max_1 = max(max_1, r-l);
        return max_1;
    }
};