题目描述
给定一个由若干 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的个数。(K是0被翻成1的个数)
这样的话就成为求一段连续相关的区间,滑动区间就派上用场了。
我们定义两个指针i,j;
j是遍历过程中,主动向右遍历的指针,在这个遍历中,需要记录 窗口里0的个数。
i是遍历过程中,被动向右遍历的指针,在这个遍历中,我们需要维护 0的个数 <= k
答案在过程中产生。
C++ 代码
class Solution {
public:
int longestOnes(vector<int>& A, int K) {
int n = A.size();
int cnt = 0 ,num = 0,res = 0;
int i = 0, j = 0;
for (;j < n;j ++){ // 滑动窗口
if (A[j] == 0) num ++; // j 是主动向右遍历
if (num > K){
while( num > K ){
if (A[i++] == 0) num --;// i 是被动向右遍历,我们要维持窗口里面0的数量保持在K的位数之内
}
}
res = max (res , j - i + 1);//最大值在遍历的过程中。
}
return res;
}
};