1、题目描述
给定一个由若干 0 和 1 组成的数组 A,我们最多可以将 K 个值从 0 变成 1 。
返回仅包含 1 的最长(连续)子数组的长度。
提示:
1 <= A.length <= 20000
0 <= K <= A.length
A[i] 为 0 或 1
2、示例
输入:A = [1,1,1,0,0,0,1,1,1,1,0], K = 2
输出:6
解释:
[1,1,1,0,0,1,1,1,1,1,1]
粗体数字从 0 翻转到 1,最长的子数组长度为 6。
输入:A = [0,0,1,1,0,0,1,1,1,0,1,1,0,0,0,1,1,1,1], K = 3
输出:10
解释:
[0,0,1,1,1,1,1,1,1,1,1,1,0,0,0,1,1,1,1]
粗体数字从 0 翻转到 1,最长的子数组长度为 10。
3、题解
基本思想:滑动窗口,这道题和424替换后的最长重复字符思路一样。
双指针+滑动窗口,维护滑动窗口[left,right]使得窗口里面最多替换k个字符使得窗口中所有字符相同,具体做法就是记录窗口中1出现次数maxcnt,那么如果right-left+1-maxcnt就是0出现的次数,如果0出现次数>k,缩小滑动窗口left++。因为right - left在整个过程是非递减的。只要right 的值加进去不满足条件,left和right就一起右滑,因为长度小于right - left的区间就没必要考虑了,所以right - left一直保持为当前的最大值。
class Solution {
public:
int longestOnes(vector<int>& nums, int k) {
int left=0,right=0,maxcnt=0;
while(right<nums.size())
{
if(nums[right]==1)
maxcnt++;
if(right-left+1-maxcnt>k)
{
if(nums[left]==1)
maxcnt--;
left++;
}
right++;
}
return right-left;
}
};