一、题目描述:487. 最大连续1的个数 II(中等)
给定一个二进制数组,你可以最多将 1 个 0 翻转为 1,找出其中最大连续 1 的个数。
示例 1:
输入:[1,0,1,1,0]
输出:4
解释:翻转第一个 0 可以得到最长的连续 1。
当翻转以后,最大连续 1 的个数为 4。
注:
输入数组只包含 0 和 1.
输入数组的长度为正整数,且不超过 10,000来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/max-consecutive-ones-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题思路
方法一、
使用双指针,left = 0,right = 0;count记录left和righ之间0的个数。left和right相当于滑动窗口
1、当count == 0的时候,right一直往右扩展,直到遇到0或者到达终点,此时如果right的下一位为1时候,right继续往右移动,直到right遇到第2个0.此时left和right之间只有一个0.计算窗口的大小,取最大值。
2、count == 1的时候,left往右收缩,直达到left遇到0,count = count-1=0。此时然后进入下一个循环
直到right到达终点
方法二、
类似动态规划,使用两个变量dp0,dp1分别记录当前位置i不含有0的最大子数组长度、含一个0的最大子数组长度。
当i= 1 dp0 = dp0 +1,dp1 = dp1+1
当i =0,,dp0 = 0,dp1 = dp0 + 1.由于dp1之前已经含有一个0了,所有再遇到0需要更新成dp0(不含0的)并且加1(相当于当前dp1又含了个0),然后dp0 初始化成0(因为dp0不能含有0)
另外定义一个变量保存dp1的最大值。
三、代码
方法一、
class Solution {
public:
int findMaxConsecutiveOnes(vector<int>& nums) {
int left = 0;
int right =0;
int count_all = 0;
int count_0 = 0;
while(right < nums.size())
{
while(right < nums.size() && count_0 == 0)
{
if(nums[right] == 0)
{
count_0++;
right++;
if(right < nums.size())
{
while(right < nums.size() && nums[right] == 1)
{
right++;
}
}
}
else
{
right ++;
}
}
//cout<<"right="<<right<<endl;
int tmp = right - left;
count_all = max(tmp,count_all);
while(count_0 == 1 && left <= right)
{
if(nums[left] == 0)
{
count_0--;
}
left++;
}
}
return count_all;
}
};
方法二、
int findMaxConsecutiveOnes(vector<int> &nums) {
int dp1 = 0;
int dp0 = 0;
int res = -1;
for (int i = 0; i < nums.size(); ++i) {
if (nums[i] == 1) {
dp1 = dp1 + 1;
dp0 = dp0 + 1;
} else {
dp1 = dp0 + 1;
dp0 = 0;
}
res = max(dp1, max(res, dp0));
}
return res;
}