【Leetcode】1004. Max Consecutive Ones III

题目地址：

https://leetcode.com/problems/max-consecutive-ones-iii/

给定一个$0-1$数组$A$，再给定一个数$k$，允许将其中的$k$个$0$变为$1$，问得到的数组中，最长的$1$子数组长度。

题目相当于在问，最长的只含不超过$k$个$0$的子数组的长度是多少，可以用双指针来做。开两个指针$i$和$j$，并保持$j\le i$，并用一个变量来存$A[j:i]$里的$0$的个数，如果个数小于等于$k$，那更新答案，否则右移$j$同时更新个数。代码如下：

class Solution {
 public:
  int longestOnes(vector<int>& a, int k) {
    int res = 0, zero_cnt = 0;
    for (int i = 0, j = 0; i < a.size(); i++) {
      zero_cnt += 1 - a[i];
      if (zero_cnt <= k)
        res = max(res, i - j + 1);
      else {
        while (zero_cnt > k) zero_cnt -= 1 - a[j++];
      }
    }

    return res;
  }
};

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