题目链接:https://leetcode-cn.com/problems/max-consecutive-ones-ii/
题目描述
给定一个二进制数组,你可以最多将 1 个 0 翻转为 1,找出其中最大连续 1 的个数。
示例 1:
输入:[1,0,1,1,0]
输出:4
解释:翻转第一个 0 可以得到最长的连续 1。
当翻转以后,最大连续 1 的个数为 4。
注:
- 输入数组只包含 0 和 1.
- 输入数组的长度为正整数,且不超过 10,000
进阶:
如果输入的数字是作为 无限流 逐个输入如何处理?换句话说,内存不能存储下所有从流中输入的数字。您可以有效地解决吗?
思路
// 滑动窗口
// 时间复杂度:O(n)空间复杂度:O(1)
class Solution {
public:
int findMaxConsecutiveOnes(vector<int>& nums) {
int zeros = 0; // 窗口内0个数
int l = 0, r = 0, ret = 0;
while(r < nums.size()){
if(nums[r] == 0)
zeros++;
while(zeros>1){
if(nums[l++] == 0)
zeros--;
}
ret = max(ret, r -l+1);
r++;
}
return ret;
}
};