Leetcode1004 最大连续1的个数III 中等(滑动窗口)

题目如图所示:

思路:
利用滑动窗口算法
cnt记录窗口内1的个数
若 窗口值 > k
用两个指针左边界收紧,直到窗口值 <= k
若 改变右边界
窗口大小更新返回值

// 个人解法
class Solution {
public:
    int longestOnes(vector<int>& A, int K) {
        int res = 0;
        for(int count = 0, l = 0, r = 0; r < A.size(); ++r) {
            if(A[r] == 1){
                ++count;
            }
            
            while (r - l + 1 - count > K){
                if(A[l++]){
                    --count;
                }
            }
            
            res = max(res, r - l + 1);
        }
        
        return res;
    }
};