题目描述:
给定一个由若干 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
解法:
class Solution {
public:
int longestOnes(vector<int>& A, int K) {
int i = 0, j = 0;
int cur = 0;
int sz = A.size();
int res = 0;
while(i < sz){
while(j < sz && cur <= K){
if(A[j] == 0){
cur++;
}
if(cur <= K){
j++;
}
}
res = max(res, j - i);
// cout<<i<<", "<<j<<","<<cur<<endl;
while(i < j && A[i] != 0){
i++;
}
cur -= 2;
i++;
}
return res;
}
};