最大连续1的个数 III
1004. 最大连续1的个数 III
给定一个二进制数组 nums
和一个整数 k
,如果可以翻转最多 k
个 0
,则返回 数组中连续 1
的最大个数 。
示例 1:
输入:nums = [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:
输入:nums = [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 <= nums.length <= 105
nums[i]
不是0
就是1
0 <= k <= nums.length
滑动窗口-统计零的数量
class Solution {
public int longestOnes(int[] nums, int k) {
int left = 0;
int right = 0;
int len = 0;
int zeros = 0;
while (right < nums.length) {
if (nums[right] == 0) zeros++;
right++;
while (zeros > k) {
if (nums[left] == 0) {
zeros--;
}
left++;
}
len = Math.max(len, right - left);
}
return len;
}
}
- 时间复杂度: O(n) ,其中 n 是数组 A 的长度。我们至多只需要遍历该数组两次(左右指针各一次)。
- 空间复杂度: O(1) ,我们只需要常数的空间保存若干变量。
滑动窗口-统计1的数量
记录窗口中1的次数,窗口长度减去1的个数就是0的个数,类似424. 替换后的最长重复字符。当窗口中需要替换的 0 的数量大于 k 时,缩小窗口。
class Solution {
public int longestOnes(int[] nums, int k) {
int left = 0, right = 0;
// 记录窗口中 1 的出现次数
int ones = 0;
// 记录结果长度
int res = 0;
// 开始滑动窗口模板
while (right < nums.length) {
// 处理信息+扩大窗口
if (nums[right] == 1) {
ones++;
}
right++;
while (right - left - windowOneCount > k) {
// 当窗口中需要替换的 0 的数量大于 k,处理信息+缩小窗口
if (nums[left] == 1) {
ones--;
}
left++;
}
// 重点。此时一定是一个合法的窗口,求最大窗口长度
res = Math.max(res, right - left);
}
return res;
}
}
- 时间复杂度: O(n) ,其中 n 是数组 A 的长度。我们至多只需要遍历该数组两次(左右指针各一次)。
- 空间复杂度: O(1) ,我们只需要常数的空间保存若干变量。