给定一个由若干 0 和 1 组成的数组 A,我们最多可以将 K 个值从 0 变成 1 。
返回仅包含 1 的最长(连续)子数组的长度。
输入:A = [1,1,1,0,0,0,1,1,1,1,0], K = 2
输出:6
输入:A = [0,0,1,1,0,0,1,1,1,0,1,1,0,0,0,1,1,1,1], K = 3
输出:10
方法一:滑动窗口法,时间复杂度O(n)
思路:
其实,将K个0转换为1之后,连续的1组成的子数组长度为最长,这个问题可以转换为:
- 在数组中取一个子数组,其中的0的个数最多为K个,怎样将这个子数组取得最长。
由于涉及到了0的个数最多为K个
这样的问题,因此这一题肯定可以用滑动窗口的方法来做。
将窗口的左端点先设定为0,右端点从0开始往右滑动,根据右端点的值以及窗口内0的个数,在右端点滑动的过程中可能有三种情况:
1.右端点处的值为1,此时只需要将右端点继续往右滑动即可。
2.右端点处的值为0且窗口内0的个数小于K个,此时将窗口右端点往右滑动,窗口内0的个数加1。
3.右端点处的值为0且窗口内的0的个数等于K个,此时先将目前窗口的长度的值减一(因为右端点多出了一个0)作为满足题意的子数组长度的极大值来更新最大值,然后左端点向右滑动直到将原有的一个0滑出窗口。
最后,由于存在某次滑动后窗口内的0的个数始终小于等于K直到数组结束
这种情况,因此在滑动结束后还需要将当前窗口的长度作为一个极大值,去更新整体的最大值。
class Solution {
public int longestOnes(int[] A, int K) {
int left = 0;
int count = 0;
int max = 0;
for(int r=0;r<A.length;r++){