【双指针】1004. 最大连续1的个数 III
给定一个由若干 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/max-consecutive-ones-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
滑动窗口解法:
- 滑动窗口[left, right]
- 滑动窗口内最多保持K个0
- 滑动窗口的移动:
- right++:右窗口右移,新增字符,统计当前窗口中0的个数
- left++ : 左窗口右移,减少字符,直到0的个数不超过K
int max(int a,int b){
return (a > b) ? a : b;
}
int longestOnes(int* A, int ASize, int K){
int right = 0;
int left = 0;
int change = 0;
int ret = 0;
while (right < ASize){
// 右窗口右移,统计当前窗口中0的个数
if (A[right] == 0){
change++;
}
right++;
// 如果当前窗口内0的个数超过K,则左窗口右移,直到0的个数不超过K
while(change > K){
if (A[left] == 0){
change--;
}
left++;
}
// 求窗口的最大值
ret = max(ret, right - left);
}
return ret;
}