LeetCode_1004_最大连续1的个数Ⅲ

题目链接

• https://leetcode.cn/problems/max-consecutive-ones-iii/

题目描述

给定一个二进制数组 nums 和一个整数 k，如果可以翻转最多 k 个 0 ，则返回 数组中连续 `1 的最大个数 。

示例 1：

输入：nums = [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：

输入：nums = [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 <= nums.length <= 10^5
• nums[i] 不是 0 就是 1
• 0 <= k <= nums.length

解题思路

在最多可以翻转k个0的情况下寻找数组nums中连续1的最大个数，等价于寻找数组nums中的最多包含k个0的最长子数组的长度

滑动窗口法

• left：控制左边界
• right：控制右边界
• count0表示滑动窗口中0的数量
• 对于每一个右边界，如果nums[right] == 0，那么count0++
  • 如果count0 > k，则当nums[left] == 0时将count0--，否则count0不变，然后将left++，直到count0 <= k
• 当前滑动窗口[left,riht]中的子数组为最多包含一个0的子数组，其长度为right - left + 1，使用当前子数组的长度更新子数组的最大长度

AC代码

class Solution {
    public int longestOnes(int[] nums, int k) {
        int left = 0;
        int right = 0;
        int ans = 0;
        int count0 = 0;

        for (right = 0; right < nums.length; right++) {
            if (nums[right] == 0) {
                count0++;
            }
            while (count0 > k) {
                if (nums[left] == 0) {
                    count0--;
                }
                left++;
            }
            ans = Math.max(ans, right - left + 1);
        }
        return ans;
    }
}