问题描述
LeetCode 1004. 最大连续1的个数 III (opens in a new tab),难度中等。
给定一个二进制数组 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
题解
滑动窗口
Solution.java
class Solution {
public int longestOnes(int[] nums, int k) {
int left = 0;
int result = 0;
int zeros = 0;
for (int right = 0; right < nums.length; right++) {
// 当遇到 0 时,计数增加
if (nums[right] == 0) {
zeros++;
}
// 调整窗口
while (zeros > k) {
if (nums[left] == 0) {
zeros--;
}
left++;
}
// 更新最大长度
result = Math.max(result, right - left + 1);
}
return result;
}
}