【Leetcode】487. Max Consecutive Ones II

题目地址：

https://leetcode.com/problems/max-consecutive-ones-ii/description/

给定一个长$n$的$01$数组$A$，允许将任意一位变为$1$，问可以得到的最长$1$子数组的长度。

双指针，对$A[i]$求最左边的$j$使得$A[j:i]$最多含$1$个$0$，则$j$关于$i$单调。代码如下：

class Solution {
 public:
  int findMaxConsecutiveOnes(vector<int>& a) {
    int cnt[2] = {0}, res = 0;
    for (int i = 0, j = 0; i < a.size(); i++) {
      cnt[a[i]]++;
      while (cnt[0] > 1) cnt[a[j++]]--;
      res = max(res, i - j + 1);
    }
    return res;
  }
};

时间复杂度$O(n)$，空间$O(1)$。