欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2024-03-06 10:36:01
...

链接

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 ) O(n) 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;
    }
};