题目
给定一个由若干 0 和 1 组成的数组 A,我们最多可以将 K 个值从 0 变成 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。
提示
- 1 <= A.length <= 20000
- 0 <= K <= A.length
- A[i] 为 0 或 1
题解
- 滑动窗口,每次移动right指针时判断当前值是否为1
- 若当前值为1,更新最大长度
- 若当前值为0,判断K的值,若K值大于0,则更新K值和最大长度;若K值为0,则更新左指针left
参考代码
class Solution {
public int longestOnes(int[] A, int K) {
int left = 0, right = 0;
int maxLen = 0;
while(right < A.length) {
if(A[right] == 0) {
if(K == 0) {
// 始终保持窗口内只有K个0
while(A[left++] == 1);
} else {
K--;
}
}
maxLen = Math.max(maxLen, right - left + 1);
right++;
}
return maxLen;
}
}