一.题目描述
这道题要我们找最大连续1的个数,看到“连续”二字,我们要想到滑动窗口的方法。滑动窗口的研究对象是一个连续的区间,这个区间需要满足某个条件。那么本题要找的是怎样的区间呢?是一个通过翻转0后得到连续1的区间,而最多可以翻转k个字符。
故要找的是包含0的个数不超过k的区间,因为如果超过k个0,即使经过翻转,该区间的1也还是不连续。
题意转化过来后,本题便不再困难。
二.思路分析
滑动窗口是在暴力解法的基础上优化过来的。本题的暴力解法就是两层for循环枚举所有的区间,找出满足条件的区间,通过比较得到最长的区间长度,结果就是数组中连续1的最大个数。
class Solution {
public:
int longestOnes(vector<int>& nums, int k)
{
int n = nums.size();
int ret = 0;
for (int left = 0; left < n; left++)
{
int zero = 0;//记录0的个数
for (int right = left; right < n; right++)
{
if (nums[right] == 0)
{
zero++;
}
//如果0的个数已经超过k,right向后枚举的区间肯定也不符合要求
if (zero > k)
{
break;
}
ret = max(ret, right - left + 1);
}
}
return ret;
}
};
要想用滑动窗口,首先要证明right没有回退的必要
如图,right从left位置出发,依次向后枚举,到图中的位置[left, right]区间内0的个数大于k,停了下来。这说明[left, right - 1]区间是满足要求的。
按照暴力枚举策略,left向右移动一步,right回退到left位置。但最终right还是会回到原来的标记处。因为通过上一轮枚举,我们可知图中大括号标记的区间都是符合条件的,而right只有在区间不满足要求时才会停下。所以right没有必要回退,留在原地即可。
那么此时[left, right]区间是否符合条件呢?答案是不一定。因为可能left跳过的是一个1, 0的数量并没有减少,也有可能跳过了一个0,区间内刚好有k个0。
当区间符合条件时,我们让right继续向后移动,接下来的步骤就和上面一样了。当区间不符合条件时,right向后枚举的区间就更不满足了,所以我们让left继续向右移动,直到区间满足要求为止。
故判断应该是一个循环语句,不能简单地只判断一次。
三.代码编写
按照滑动窗口的模版,找到各个条件即可。当枚举的情况满足要求时应该更新结果。什么时候满足要求呢?
1.进窗口之后,zero>=k,符合要求
2.进窗口之后,zero<k,经过若干次出窗口操作后,zero=k ,满足要求
故更新结果应放在整个循环的最后面
class Solution {
public:
int longestOnes(vector<int>& nums, int k) {
int n =nums.size();
int zero = 0;//记录窗口内0的个数
int left = 0, right = 0;
int ret = 0;
while (right < n)
{
//进窗口
if (nums[right] == 0)
{
zero++;
}
//判断
while (zero > k)
{
//出窗口
if (nums[left] == 0)
{
zero--;
}
left++;
}
//更新结果
ret = max(ret, right - left + 1);
right++;
}
return ret;
}
};
时间复杂度O(n),相比于暴力枚举的O(n^2)提升了不少。