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]

示例 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的个数：

class Solution {
    public int longestOnes(int[] nums, int k) {
        int n = nums.length;
        int res = 0;
        int cnt = 0;
        int left=0,right=0;
        while(right < n){
            if(nums[right] == 0){
                res++;
            }
            while(res > k){
                res = 0;//下一个窗口中0的个数重置为0
                left++;
            }
            cnt = Math.max(cnt, right-left+1);
            right++;
        }
        return cnt;
       
    }
}

结果答案不对，因为我将下一个窗口中0的个数重置为0，重新统计0的个数。

没有理解模板中的，“此时需要一直移动左指针，直至找到一个符合题意的区间”。

class Solution {
    public int longestOnes(int[] nums, int k) {
        int n = nums.length;
        int res = 0;
        int cnt = 0;
        int left=0,right=0;
        while(right < n){
            if(nums[right] == 0){
                res++;
            }
            while(res > k){
                if(nums[left] == 0){ //这句是关键，此时需要一直移动左指针，直至找到下一个符合题意的区间。
                    res--;
                }
                left++;
            }
            cnt = Math.max(cnt, right-left+1);
            right++;
        }
        return cnt;
       
    }
}