例题:给定一个二进制数组 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
解题思路:重点是题意转换,把【最多可以把K个0变成1,求仅包含1的最长子数组的长度】转换为【找出一个最长的子数组,该子数组内最多允许有K个0】
经过上面的题意转换,我们可知本题是求最大连续子区间,可以使用滑动窗口方法。滑动窗口的限制条件是:窗口内最多有K个0。
可以使用滑动窗口模板解决。
解题思路:维护一个变量cnt表示闭合区间[left,right]内有多少个0,通过滑窗思路解决
代码思路:
- 使用left和right两个指针,分别指向滑动窗口的左右边界。
- right主动右移:right指针每次移动一步。当nums[right]为0,说明滑动窗口内增加了一个0;
- left被动右移:判断此时窗口内0的个数,如果超过了K,则left指针被迫右移,直至窗口内的0的个数小于等于K为止。
- 滑动窗口的最大长度值就是最终答案
代码如下:
public static int GetMaxSubArraryLength(int[] nums, int cnt) {
int ans = 0;
int left = 0, right = 0;
int len = nums.length;
while (right < len) {
if (nums[right] == 0) {
cnt--;
}
while (cnt < 0) {
if (nums[left] == 0) {
cnt++;
}
left++;
}
if (cnt == 0) {
ans = Math.max(ans, right - left + 1);
}
right++;
}
return ans;
}
模板:
def findSubArray(nums):
N = len(nums) # 数组/字符串长度
left, right = 0, 0 # 双指针,表示当前遍历的区间[left, right],闭区间
sums = 0 # 用于统计 子数组/子区间 是否有效,根据题目可能会改成求和/计数
res = 0 # 保存最大的满足题目要求的 子数组/子串 长度
while right < N: # 当右边的指针没有搜索到 数组/字符串 的结尾
sums += nums[right] # 增加当前右边指针的数字/字符的求和/计数
while 区间[left, right]不符合题意:# 此时需要一直移动左指针,直至找到一个符合题意的区间
sums -= nums[left] # 移动左指针前需要从counter中减少left位置字符的求和/计数
left += 1 # 真正的移动左指针,注意不能跟上面一行代码写反
# 到 while 结束时,我们找到了一个符合题意要求的 子数组/子串
res = max(res, right - left + 1) # 需要更新结果
right += 1 # 移动右指针,去探索新的区间
return res
滑动窗口模板还可再精简,将双层循环转换成单层循环解决:
public static int GetMaxSubArraryLen(char[] nums, int cnt)
{
int len = nums.length;
int left = 0, right = 0;
// 滑动窗口优先扩大,尝试性扩大,若不符合要求,左指针被迫右移,若符合窗口扩大条件左指针不动,右指针持续扩大
// 滑窗具有记忆功能,一旦扩大就不会再缩小尺寸
while (right < len) {
if (nums[right++] == 0)
cnt--;
if (cnt < 0 && nums[left++] == 0)
cnt++;
}
return right - left; // 注意计算最大长度处理,不用 +1
}