LeetCode【1004】最大连续1的个数 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的次数
滑动窗口的题目，控制一个左右窗口指针，向右移动右窗口，zero用于控制0变成1的个数，每用一次zero+1，直到zero>K时，将左窗口移动到离左窗口最近的0的位置，保证使用完K的情况下，最大的连续1的个数，记录为max
例：1,1,1,0,0,0,1,1,1,1,0 K = 2

1、【1】,1,1,0,0,0,1,1,1,1,0
2、【1,1】,1,0,0,0,1,1,1,1,0
3、【1,1,1】,0,0,0,1,1,1,1,0
4、【1,1,1,0】,0,0,1,1,1,1,0 zero++
5、【1,1,1,0,0】,0,1,1,1,1,0 zero++
6、【1,1,1,0,0,0】,1,1,1,1,0 zero==3 > K 此时左指针++，一直移动到下一个0，的位置
7、1,1,1,0,【0,0】,1,1,1,1,0
8、1,1,1,0,【0,0,1】,1,1,1,0
9、1,1,1,0,【0,0,1,1】,1,1,0
10、1,1,1,0,【0,0,1,1,1】,1,0
11、1,1,1,0,【0,0,1,1,1,1】,0
12、1,1,1,0,【0,0,1,1,1,1,0】 zero ==3 > K 此时左指针++
13、1,1,1,0,0,【0,1,1,1,1,0】
总之，窗口内【】最多为K个0时，窗口最大的长度，即为所求

Java代码

public static int longestOnes(int[] A, int K) {
        int left = 0, right = 0;
        int max = 0;
        int zero = 0;
        while (right != A.length) {
            if (A[right++] == 0) {
                zero++;
            }
            //判定条件，0的个数大于K
            while (zero > K) {
                if (A[left++] == 0) {
                    --zero;
                }
            }
            int count = right - left;
            max = max > count ? max : count;
        }
        return max;
    }