最大连续1的个数 III

最大连续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。

解题思路：运用滑动窗口解决，维护 left 和 right 两个边界，并且用一个变量来记录当前窗口中0的数量，当0的数量大于限定的K值后则要不断增大 left 来减小当前窗口中0的数量

class Solution {
    public int longestOnes(int[] A, int K) {
        int l = 0, r = 0, k = 0, len = A.length, ans = 0;
        
        while(r < len) {
            if(A[r] == 0) {
                k++;
            }
            
            while(k > K) {
                if(A[l] == 0) {
                    k--;
                }
                
                l++;
            }
            
            ans = Math.max(ans, r - l + 1);
            r++;
        }
        
        return ans;
    }
}