力扣---1004. 最大连续1的个数 III

发布时间 2023-04-29 16:51:41作者: Owlwu

给定一个二进制数组 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 <= 105
nums[i] 不是 0 就是 1
0 <= k <= nums.length

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/max-consecutive-ones-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


 

经典滑动窗口题。右边界一直移动,左边界视情况收缩,维护窗口中 0 的个数始终等于 k 即可。

class Solution {
    public int longestOnes(int[] nums, int k) {
        int len = nums.length;
        int left = 0;
        int right = 0;
        // 先找到第一个窗口
        for (int count = 0; count < k; right++) {
            // nums 中 0 的个数小于等于 k,即数组中所有的数都可以被翻转。
            if (right >= len) {
                return len;
            } else if (nums[right] == 0) {
                count ++;
            }
        }
        // 第一个窗口的长度即是第一个答案。
        int max = right;
        while (right < len) {
            // 右指针指向等于 1 时,可以加入到窗口,right ++
            if (nums[right] == 1) {
                right ++;
                max = Math.max(right - left, max);
            } else if (nums[left] == 1) { 
                // 此时窗口中 0 的个数大于 k ,需要移动左指针保证 0 的个数等于 k 。
                left ++;
            } else {
                left ++;
                right ++;
            }
        }
        return max;
    }
}