每日OJ题_算法_滑动窗口③_力扣1004. 最大连续1的个数 III

目录

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

解析及代码

---

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

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

难度 中等

给定一个二进制数组 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

class Solution {
public:
    int longestOnes(vector<int>& nums, int k) {
 
    }
};

---

解析及代码

此题可以转化为找出最长子数组，0的个数不超过K个，就和前两篇滑动窗口的题目一样了。

设置计数器，计算0的个数

首先数组不能越界，所以最外层循环进入条件为right<n（数组长度）

滑动窗口四步走：

1. 进窗口，遇到0计数器++，否则忽略（ ps：滑动窗口的进窗口操作是载入数据，而不是单纯的移动right指针，如果只移动了right而没有对数据进行处理，则不算进窗口。）
2. 判断，0个数是否大于 题目中的k （不是等于而是大于，这是因为如果在计数器的0个数等于k时就停止进窗口，并更新状态，那么就会漏情况。比如k为2 那么数组 1  1  0  0  1  1  1 1 这个数组中最长子串的长度就是8，如果是在0等于2个时就停下，那么更新的结果就会是4。）
3. 出窗口，left移动，遇到0时计数器--，直到计数器的0个数等于k个，符合题目要求为止。
4. 更新结果，无论是否出窗口都更新状态。只要用个max函数来判断此时的子串是否是最大长度即可。

class Solution {
public:
    int longestOnes(vector<int>& nums, int k) {
    // 时间O（N）， 空间O（1）
    int n = nums.size(), ret = 0, left = 0, right = 0, zero = 0;
    while(right < n)
    {
        if(nums[right] == 0) 
        {
            zero++; // 进窗⼝
        }
        while(zero > k) // 判断
        {
            if(nums[left++] == 0) 
            {
                zero--; // 出窗⼝
            }
        }
        ret = max(ret, right - left + 1); // 更新结果
        ++right;
    }
    return ret;
    }
};