LeetCode 每日一题1004. 最大连续1的个数 III

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

方法一：滑动窗口

解题思路

• 依然用双指针 [left, right) 模拟滑动窗口
• right 指针向右遍历，当遇到 K 个 0 时，计算结果值 ret = max(ret, right - left) 并 移动 left 指针到下一个 0 的右边。

参考代码

public int longestOnes(int[] A, int K) {
    int n = A.length;
    int ret = 0;
    int left = 0, right = 0, count = 0;
    while (right < n) {
        if (A[right] == 0) {
            if (count < K) {
                count++;
            } else {
                ret = Math.max(ret, right - left);
                do {
                    left++;
                } while (A[left - 1] == 1);
            }
        }
        right++;
    }
    return Math.max(ret, right - left);
}

执行结果

• 时间复杂度：O(n)
• 空间复杂度：O(1)