Leetcode 1004. 最大连续1的个数 III (双指针)

Description

Solution
显然满足双指针中的单调性，即左指针+1的情况下，右指针(非严格)递增

Code

class Solution {
public:
    int longestOnes(vector<int>& A, int K) {
        int l = 0, r = 0, res = 0;
        int cnt = K, n = A.size();
        if(!K) {
            int tmp = 0;
            for(int i = 0;i < n;++i) {
                if(!A[i]) continue;
                tmp = 0;
                while(i < n && A[i]) {
                    tmp++;i++;
                }res = max(res, tmp);
            }return res;
        }
        if(!A[0]) {
            cnt--;
        }
        while(l < n) {
            while(r + 1 < n) {
                if(A[r+1]) {
                    r++;
                } else {
                    if(cnt) cnt--,r++;
                    else break;
                }
            }
            res = max(res, r - l + 1);
            if(!A[l]) cnt++;
            l++;
        }return res;
    }
};