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

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

思路：

用滑动窗口法

c++

class Solution {
public:
    int longestOnes(vector<int>& nums, int k) {
        int left = 0, right = 0,min_len = 0;

        map<int,int> window; // 记录窗口内 1 与 0的数量
        window[0] = 0;
        window[1] = 0;
        while(right<nums.size()) {
            int c = nums[right];
            right++;
            window[c]++;

            if(window[0] <= k) {
                min_len = min_len<(window[1] + window[0])?(window[1] + window[0]):min_len;
            }
            // 窗口内 0的数量超过 k ，意味着窗口内元素无法全 1
            while(window[0]>k) {
                int d = nums[left];
                left++;
                window[d]--;
            }
        }

        return min_len;
    }

};