[LeetCode] 487. Max Consecutive Ones II

Given a binary array, find the maximum number of consecutive 1s in this array if you can flip at most one 0.

Example 1:

Input: [1,0,1,1,0]
Output: 4
Explanation: Flip the first zero will get the the maximum number of consecutive 1s.
    After flipping, the maximum number of consecutive 1s is 4.

Note:

• The input array will only contain 0 and 1.
• The length of input array is a positive integer and will not exceed 10,000

Follow up:
What if the input numbers come in one by one as an infinite stream? In other words, you can't store all numbers coming from the stream as it's too large to hold in memory. Could you solve it efficiently?

最大连续 1 的个数 II。

给定一个二进制数组，你可以最多将 1 个 0 翻转为 1，找出其中最大连续 1 的个数。

思路是滑动窗口。这道题需要动动脑筋，题意可以转化为你需要找一个最长的子数组，里面最多只能有一个 0。这道题我们需要一个 queue 记录所有遍历到的 0 的位置，而且 queue 的 size 不能大于 1。当 queue.size() > 1 的时候，我们需要结算一下如果把上一个找到的 0 替换成 1，能凑成的满足题意的子串长度是多少。结算完毕之后，需要把左指针移动到当前的 0 的右边，这也是使用队列的巧妙之处。

时间O(n)

空间O(n)

Java实现

class Solution {
    public int findMaxConsecutiveOnes(int[] nums) {
        int max = 0;
        // 保持queue中只有一个1
        // 等价于滑动窗口找子串，子串内最多只能有一个1
        int k = 1;
        Queue<Integer> queue = new LinkedList<>();
        for (int l = 0, h = 0; h < nums.length; h++) {
            if (nums[h] == 0) {
                queue.add(h);
            }
            if (queue.size() > k) {
                l = queue.poll() + 1;
            }
            max = Math.max(max, h - l + 1);
        }
        return max;
    }
}

 

我再提供一个类似 76 题的模板写法，这个写法可以直接套用到 1004 题上。

时间O(n)

空间O(1)

Java实现

class Solution {
    public int findMaxConsecutiveOnes(int[] nums) {
        int res = 0;
        int count = 0;
        int start = 0;
        int end = 0;
        while (end < nums.length) {
            if (nums[end] == 0) {
                count++;
            }
            end++;
            while (count > 1) {
                if (nums[start] == 0) {
                    count--;
                }
                start++;
            }
            res = Math.max(res, end - start);
        }
        return res;
    }
}

 

sliding window相关题目

LeetCode 题目总结