题目描述:
给定一个由若干 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。
思路分析:
一看到最大两字,就先想到滑动窗口。然后再读题意,可知就是用滑动窗口解决。
我们可以理解为,滑动窗口内最多有K个0,求滑动窗口最大的长度。
接下来代码就简单啦!
class Solution {
public int longestOnes(int[] A, int K) {
int count=0;//统计0的个数
int left=0;//滑动窗口区间左端点
int right=0;//滑动窗口区间右端点
int res=0;//最终结果
//滑动窗口表示的区间为[left,right),左闭右开
while(right<A.length){
if(A[right++]==0){//窗口扩充一个元素,如果为0则count++;
count++;
}
while(count>K){//当窗口内0的个数超过k时候,开始收缩窗口
if(A[left++]==0){//如果刚滑出窗口的元素是0,则count--;
count--;
}
}
//此时count<=K,保存窗口的最大宽度
res=Math.max(res,right-left);
}
return res;
}
}