给定一个由若干 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。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/max-consecutive-ones-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:使用滑动窗口,统计窗口中0的个数,如果没有大于k,则可以继续向右扩大,如果大于k,则左边要进行缩小,
class Solution {
public:
int longestOnes(vector<int>& A, int K) {
int left = 0;
int right = 0;
int count = 0;
int ans = 0;
while(right < A.size())
{
if(A[right] == 0)//继续扩张
++count;
if(count > K)//准备缩小
{
if(A[left] == 0)//左边也是0 那么0的数量减少
--count;
++left;
}
ans = max(ans,right - left + 1);//统计区间长度
++right;
}
return ans;
}
};