LeetCode 75系列 - 1004. 最大连续1的个数 III

题目描述

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

示例 1：

1
2
3
4
输入：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：

1
2
3
4
输入：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

题解

这道题的思路是使用滑动窗口，然后寻找最长为1的子数组，同时需要考虑为0的时候可以反转为1，也就是当0的个数在k以内，可以算在最长为1的数组里。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution {
public:
    int longestOnes(vector<int>& nums, int k) {
        int max = {};
        int left{};
        int num{};
        for (int right = 0; right < nums.size(); ++right) {
            // 当前位置为0时，记录0的个数
            if (nums[right] == 0) {
                num++;
            }
            // 当0的个数超过k时，移动left位置，同时判断当前位置是否为0，如果是，则减一
            if (num > k) {
                if (nums[left++] == 0)
                    num--;
            }
            // right - left + 1为最长为1的长度（含k个0）
            max = std::max(max, right - left + 1);
        }
        return max;
    }
};