leetcode 1004. 最大连续1的个数 III(Max Consecutive Ones III)

目录

• 题目描述：
• 示例 1：
• 示例 2：
• 解法：

题目描述：

给定一个由若干 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]
    粗体数字从 0 翻转到 1，最长的子数组长度为 6。

示例 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 翻转到 1，最长的子数组长度为 10。

提示：

• 1 <= A.length <= 20000
• 0 <= K <= A.length
• A[i] 为 0 或 1

---

解法：

class Solution {
public:
    int longestOnes(vector<int>& A, int K) {
        int i = 0, j = 0;
        int cur = 0;
        int sz = A.size();
        int res = 0;
        while(i < sz){
            while(j < sz && cur <= K){
                if(A[j] == 0){
                    cur++;
                }
                if(cur <= K){
                    j++;
                }
            }
            res = max(res, j - i);
            // cout<<i<<", "<<j<<","<<cur<<endl;
            while(i < j && A[i] != 0){
                i++;
            }
            cur -= 2;
            i++;
        }
        return res;
    }
};