2021年2月19日 Leetcode每日一题:1004. 最大连续1的个数 III

最大连续1的个数Ⅲ

1.题目描述

给定一个由若干 0 和 1 组成的数组 A,我们最多可以将 K 个值从 0 变成 1 。

返回仅包含 1 的最长(连续)子数组的长度。

2.示例

示例 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。

3.读题

这一题题意简单明了。数组由数字1和0组成,我们可以将其中任意K个0变成1。我们要找出,将数组中的K个0变成1之后,可以在数组中组成的最长的由连续的1组成的子数组的长度。

4.思路

其实,将K个0转换为1之后,连续的1组成的子数组长度为最长,这个问题可以转换为:

  • 在数组中取一个子数组,其中的0的个数最多为K个,怎样将这个子数组取得最长。

由于涉及到了0的个数最多为K个这样的问题,因此这一题肯定可以用滑动窗口的方法来做。

将窗口的左端点先设定为0,右端点从0开始往右滑动,根据右端点的值以及窗口内0的个数,在右端点滑动的过程中可能有三种情况:

1.右端点处的值为1,此时只需要将右端点继续往右滑动即可。
2.右端点处的值为0且窗口内0的个数小于K个,此时将窗口右端点往右滑动,窗口内0的个数加1。
3.右端点处的值为0且窗口内的0的个数等于K个,此时先将目前窗口的长度的值减一(因为右端点多出了一个0)作为满足题意的子数组长度的极大值来更新最大值,然后左端点向右滑动直到将原有的一个0滑出窗口。

最后,由于存在某次滑动后窗口内的0的个数始终小于等于K直到数组结束这种情况,因此在滑动结束后还需要将当前窗口的长度作为一个极大值,去更新整体的最大值。

4.代码

class Solution {
    public int longestOnes(int[] A, int K) {
        int left = 0;
        int count = 0;
        int max = 0;
        for(int r=0;r<A.length;r++){
            if(A[r]==1){
                continue;
            }else if(A[r]==0&&count<K){
                count++;
            }else{
                max = Math.max(max,(r-left));
                while(A[left]!=0) left++;
                left++;
            }
        }
        max = Math.max(max,(A.length-left));
        return max;
    }
}