快乐写代码

T1004、最大连续1的个数

2020-08-28  本文已影响0人  上行彩虹人

给定一个由若干 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

维护一个滑动窗口。

   public int longestOnes(int[] A, int K) {
        int left = 0, right = 0, count = 0;
        int res = 0;
        while(right < A.length){
            if(A[right] == 0)
                count++;
            while(count > K){
                if(A[left] == 0)
                    count--;
                left++;
            }
            right++;
            res = Math.max(res, right - left);
        }
        return res;
    }
上一篇下一篇

猜你喜欢

热点阅读