水果成篮
你正在探访一家农场,农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示,其中 fruits[i] 是第 i 棵树上的水果 种类 。
你想要尽可能多地收集水果。然而,农场的主人设定了一些严格的规矩,你必须按照要求采摘水果:
你只有 两个 篮子,并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
你可以选择任意一棵树开始采摘,你必须从 每棵 树(包括开始采摘的树)上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次,你将会向右移动到下一棵树,并继续采摘。
一旦你走到某棵树前,但水果不符合篮子的水果类型,那么就必须停止采摘。
给你一个整数数组 fruits ,返回你可以收集的水果的 最大 数目。
限制
1 <= fruits.length <= 100000
0 <= fruits[i] < fruits.length
示例
示例 1:
输入:fruits = [1,2,1]
输出:3
解释:可以采摘全部 3 棵树。
示例 2:
输入:fruits = [0,1,2,2]
输出:3
解释:可以采摘 [1,2,2] 这三棵树。
如果从第一棵树开始采摘,则只能采摘 [0,1] 这两棵树。
示例 3:
输入:fruits = [1,2,3,2,2]
输出:4
解释:可以采摘 [2,3,2,2] 这四棵树。
如果从第一棵树开始采摘,则只能采摘 [1,2] 这两棵树。
示例 4:
输入:fruits = [3,3,3,1,2,1,1,2,3,3,4]
输出:5
解释:可以采摘 [1,2,1,1,2] 这五棵树。
具体含义
树上一定有果子,只是种类不同。
你一定得摘果子,每次遇到新种类还得摘一种新的。
篮子最多放两种,数量不限。
例如例二,0代表树上果子的种类不在重复列表中,摘了之后单独占一个格子,故01之后放不了2或者3之类的。
如例三,1摘了之后占一个格子,2再占一个,这个时候3出现,为了摘更多种类的,放不下了,从1开始就最多2棵树。从2开始,占一个,到3这时摘跟2重复的交集的那种就行,后续的2就仍在果篮的种类范围内,便可继续摘。
或者换句话说,查找包含元素种类不超过两种的最长子串,返回子串长度。
思路
滑动窗口+哈希
思路与力扣76. 最小覆盖子串几乎相同,只是那个题操作的是字符串。
左指针和右指针起始位置相同,右指针移动,种类表(哈希表,也就是篮子)记录该类型的出现次数,同时出现次数满一次就记类型出现了。
当超过两种就放不下了,此时移动左指针,移动就从篮子里取走,种类表对应减一。种类表对应为0时就取光了,种类计数减一,使得跳出循环。
这时计算滑动窗口长度并与最大值进行比较和保存,完成后再移动右指针,继续循环。
时间复杂度为 O(n),其中 n 是 fruits 数组的长度,空间复杂度为 O(1),因为 type 数组的大小是固定的,不随 fruits 数组的大小变化。
(leetcode测试例子有数很大的所以type很大,参见限制)
class Solution {
public:
int totalFruit(vector<int>& fruits) {
int left = 0, right = 0, maxLen = 0;
int type[100001] = {0}; // 假设水果编号从 1 到 100001 题目数字大小限制
int count = 0; // 记录当前窗口中不同水果的类型数
while (right < fruits.size()) {
type[fruits[right]]++; // 增加当前水果的计数
if (type[fruits[right]] == 1) { // 如果是新的水果类型
count++; // 增加水果类型计数
}
// 当有超过两种类型的水果时,移动 left 指针
while (count > 2) {
type[fruits[left]]--;
if (type[fruits[left]] == 0) { // 如果类型计数为 0,说明这种水果已经全部移出窗口
count--; // 减少水果类型计数
}
left++; // 移动 left 指针向前
}
// 更新最长长度
maxLen = max(maxLen, right - left + 1);
right++; // 移动 right 指针向前
}
return maxLen;
}
};