滑动窗口框架
/* 滑动窗口算法框架 */
void slidingWindow(string s, string t) {
unordered_map<char, int> need, window;
for (char c : t) need[c]++;
int left = 0, right = 0;
int valid = 0;
while (right < s.size()) {
// c 是将移入窗口的字符
char c = s[right];
// 右移窗口
right++;
// 进行窗口内数据的一系列更新
...
/*** debug 输出的位置 ***/
printf("window: [%d, %d)\n", left, right);
/********************/
// 判断左侧窗口是否要收缩
while (window needs shrink) {
// d 是将移出窗口的字符
char d = s[left];
// 左移窗口
left++;
// 进行窗口内数据的一系列更新
...
}
}
}
209.长度最小的子数组
问题:给定一个含有 n 个正整数的数组和一个正整数 target 。
找出该数组中满足其和 ≥ target 的长度最小的 连续子数组 [numsl, numsl+1, …, numsr-1, numsr] ,并返回其长度。如果不存在符合条件的子数组,返回 0 。
示例:
输入:target = 7, nums = [2,3,1,2,4,3]
输出:2
解释:子数组 [4,3] 是该条件下的长度最小的子数组。
输入:target = 4, nums = [1,4,4]
输出:1
输入:target = 11, nums = [1,1,1,1,1,1,1,1]
输出:0
思路:
- 滑动窗口
class Solution {
public int minSubArrayLen(int target, int[] nums) {
int len = nums.length;
int l = 0, r = 0;
int subNum = 0, minSubArr = Integer.MAX_VALUE;
while(r < len){
subNum += nums[r];
while(subNum >= target){
minSubArr = Math.min(minSubArr, r - l + 1);
subNum -= nums[l++];
}
//因为循环内涉及长度计算,所以需要计算完再右移,也可以先右移,计算长度时对应的少加1
r++;
}
return minSubArr == Integer.MAX_VALUE ? 0 : minSubArr;
}
}
- 暴力法需要枚举数组中每个下标
i
作为子数组的开始下标,这是外层循环,内层循环需要对每个以i
开始的子数组进行计算,需要找到大于等于下标i的最小下标j,使得nums[i]到nums[j]的元素累加和 >= target
,也就是使得子数组nums[i...j]的累加和>=target
。寻找j的过程的时间复杂度为O(n)。我们可以用前缀和+二分查找将内存循环的时间复杂度进行优化。
class Solution {
public int minSubArrayLen(int target, int[] nums) {
int len = nums.length;
int[] preSum = new int[len + 1];
for(int i = 1; i < preSum.length; i++){
preSum[i] = preSum[i - 1] + nums[i - 1];
}
int minSubArr = Integer.MAX_VALUE;
for(int i = 0; i < preSum.length; i++){
int l = i, r = preSum.length - 1;
if(preSum[r] - preSum[l] < target) break;
while(l <= r){
int m = (l + r) >> 1;
//子数组nums[i...m]的累加和不符合条件,在左边区间找。
if(preSum[m] - preSum[i] < target){
l = m + 1;
} else {
//符合条件,则寻找最小的那个j
r = m - 1;
}
}
minSubArr = Math.min(minSubArr, l - i);
}
return minSubArr == Integer.MAX_VALUE ? 0 : minSubArr;
}
}
//这种写法,求preSum[j] - preSum[i] >= target nums[i-1 ... j]的累加和>=target
//也就是说 求preSum[j] >= preSum[i] + target j尽可能小 类似于确定边界
class Solution {
public int minSubArrayLen(int target, int[] nums) {
int len = nums.length;
int[] preSum = new int[len + 1];
for(int i = 1; i < preSum.length; i++){
preSum[i] = preSum[i - 1] + nums[i - 1];
}
int minSubArr = Integer.MAX_VALUE;
for(int i = 0; i < preSum.length; i++){
int l = i, r = preSum.length - 1;
if(preSum[r] - preSum[l] < target) break;
int tar = preSum[i] + target;
while(l <= r){
int m = (l + r) >> 1;
if(preSum[m] < tar){
l = m + 1;
} else {
r = m - 1;
}
}
minSubArr = Math.min(minSubArr, l - i);
}
return minSubArr == Integer.MAX_VALUE ? 0 : minSubArr;
}
}
904.水果成篮
问题:你正在探访一家农场,农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示,其中 fruits[i] 是第 i 棵树上的水果 种类 。
你想要尽可能多地收集水果。然而,农场的主人设定了一些严格的规矩,你必须按照要求采摘水果:
-
你只有 两个 篮子,并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
-
你可以选择任意一棵树开始采摘,你必须从 每棵 树(包括开始采摘的树)上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次,你将会向右移动到下一棵树,并继续采摘。
-
一旦你走到某棵树前,但水果不符合篮子的水果类型,那么就必须停止采摘。
给你一个整数数组 fruits ,返回你可以收集的水果的 最大 数目。
示例:
输入:fruits = [1,2,1]
输出:3
解释:可以采摘全部 3 棵树。
输入:fruits = [0,1,2,2]
输出:3
解释:可以采摘 [1,2,2] 这三棵树。
如果从第一棵树开始采摘,则只能采摘 [0,1] 这两棵树。
输入:fruits = [1,2,3,2,2]
输出:4
解释:可以采摘 [2,3,2,2] 这四棵树。
如果从第一棵树开始采摘,则只能采摘 [1,2] 这两棵树
输入:fruits = [3,3,3,1,2,1,1,2,3,3,4]
输出:5
解释:可以采摘 [1,2,1,1,2] 这五棵树。
思路:滑动窗口
这道题其实就是求最多包含两种元素的最长连续子数组。
- 不使用额外空间版本
class Solution {
public int totalFruit(int[] fruits) {
int l = 0, r = 0; //r表示滑动窗口的右边界,l即表示滑动窗口的左边界,又表示篮子中的一种水果
int f1 = -1; //篮子中另一种水果
int maxNum = -1; //记录篮子中最大的水果数
while(r < fruits.length){
if(f1 == -1 && fruits[l] != fruits[r]){
//第一次找到了第二种水果
f1 = r;
}
//找到了第三种水果
if(fruits[r] != fruits[l] && fruits[r] != fruits[f1]){
//此时r指向的为第三种水果,所以篮子中两种水果的数为r-l
maxNum = Math.max(maxNum, r - l);
//类似于将篮子中的水果按顺序拿出来,只留下一种水果。
//子数组中只有两种元素,而且最后一个元素fruits[r-1]肯定是留在篮子里的。
//所以我选择从子数组末尾开始遍历,找到第一个与 fruits[r-1] 不相等的元素,
//然后将这个元素及其前面的元素都从篮子中拿出去,然后继续摘水果
f1 = r-1;
l = r-1;
//找到第一个与 fruits[r-1] 不相等的元素
while(fruits[f1] == fruits[l]){
f1--;
}
//将这个元素及其前面的元素都从篮子中拿出去
l = f1 + 1;
//此时的r所指向的就是篮子中的第二种水果。
f1 = r;
}
r++;
}
return Math.max(r - l, maxNum);
}
}
- 使用数组版本 感觉更容易理解
class Solution {
public int totalFruit(int[] fruits) {
int l = 0, r = 0;
int[] box = new int[fruits.length];
int res = 0, size = 0;
while(r < fruits.length){
if(box[fruits[r]] == 0){
//篮子中水果种类加1
size++;
}
//篮子中fruits[r]种类的水果数量加1
box[fruits[r]]++;
//篮子中水果种类超过2种,将多余的水果丢出篮子,收缩窗口
while(size > 2){
box[fruits[l]]--;
if(box[fruits[l]] == 0){
size--;
}
l++;
}
//更新答案
res = Math.max(res, r - l + 1);
//扩大窗口
r++;
}
return res;
}
}
整理思路,记录博客,以便复习。若有误,望指正~