题目描述:
给定一个由若干 0 和 1 组成的数组 A,我们最多可以将 K 个值从 0 变成 1 。
返回仅包含 1 的最长(连续)子数组的长度。
示例 1:
输入: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。
示例 2:
输入: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。
提示:
1 <= A.length <= 20000
0 <= K <= A.length
A[i] 为 0 或 1
方法1:双指针,滑动窗口
主要思路:
(1)使用双指针,一个指向窗口的左侧 left,一个指向窗口的右侧 right,初始值都为 0 ;
(2)先将右侧指针尽量右移,知道不满足条件,使用 right-left 获得当前窗口的大小(中间一直保留最大窗口的大小),然后再将做侧指针移动一位,判断当前窗口的变换;
(3)对于第二步,具体的在移动右侧指针时候,由于这里规定了 K 次可以调换 0 为 1,故只要当前值为 1 ,或为0的时候,但K大于0,右侧指针都可以向后移动,直到不满足要求,既 当前值 为0 且 K为0了;
(4)对于第二步,具体的在移动左侧指针的时候,若是当前值为 1,则只移动left,若是当前值为为0,则同时移动左侧指针和右侧指针,这里相当于释放出来一个K值,让右侧指针可以越过当前的不能继续移动的一个位置;
class Solution {
public:
int longestOnes(vector<int>& A, int K) {
//处理特殊情形
if(A.size()<=K)
return A.size();
//左右侧指针和保存最大窗口长度的变量
int left=0;
int right=0;
int max_len=0;
//窗口没有越界
while(right<A.size()){
while(right<A.size()){ //移动右侧指针
if(A[right]==1){//当前值为1,直接移动右侧指针
++right;
}
else if(K>0){//当前值为 0,但K大于0,说明可以跳过该位置
++right;
--K;
}
else{//不能够再移动右侧指针
break;
}
}
max_len=max(max_len,right-left);//获得当前窗口的大小,并更新可能的最大窗口
if(A[left]==1){//在当前值为 1 时,直接移动左侧指针
++left;
}
else {//在当前值为 0,同时移动左侧指针和右侧指针,让右侧指针跳过当前 位置
++left;
++right;
}
}
return max_len;
}
};