Skip to content

最大连续1的个数 III

约 603 字大约 2 分钟

滑动窗口哈希表

2025-02-25

1004. 最大连续1的个数 III

给定一个二进制数组 nums 和一个整数 k,如果可以翻转最多 k0 ,则返回 数组中连续 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

滑动窗口-统计零的数量

class Solution {
    public int longestOnes(int[] nums, int k) {
        int left = 0;
        int right = 0;
        int len = 0;
        int zeros = 0;

        while (right < nums.length) {
            if (nums[right] == 0) zeros++;
            right++;

            while (zeros > k) {
                if (nums[left] == 0) {
                    zeros--;
                }
                left++;
            }

            len = Math.max(len, right - left);
        }

        return len;
    }
}
  • 时间复杂度: O(n)O(n) ,其中 nn 是数组 AA 的长度。我们至多只需要遍历该数组两次(左右指针各一次)。
  • 空间复杂度: O(1)O(1) ,我们只需要常数的空间保存若干变量。

滑动窗口-统计1的数量

记录窗口中1的次数,窗口长度减去1的个数就是0的个数,类似424. 替换后的最长重复字符。当窗口中需要替换的 0 的数量大于 k 时,缩小窗口。

class Solution {
    public int longestOnes(int[] nums, int k) {
        int left = 0, right = 0;
        // 记录窗口中 1 的出现次数
        int ones = 0;
        // 记录结果长度
        int res = 0;

        // 开始滑动窗口模板
        while (right < nums.length) {
            // 处理信息+扩大窗口
            if (nums[right] == 1) {
                ones++;
            }
            right++;

            while (right - left - windowOneCount > k) {
                // 当窗口中需要替换的 0 的数量大于 k,处理信息+缩小窗口
                if (nums[left] == 1) {
                    ones--;
                }
                left++;
            }
            
            // 重点。此时一定是一个合法的窗口,求最大窗口长度
            res = Math.max(res, right - left);
        }
        return res;
    }
}
  • 时间复杂度: O(n)O(n) ,其中 nn 是数组 AA 的长度。我们至多只需要遍历该数组两次(左右指针各一次)。
  • 空间复杂度: O(1)O(1) ,我们只需要常数的空间保存若干变量。