你正在探访一家农场,农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示,其中 fruits[i] 是第 i 棵树上的水果 种类 。
你想要尽可能多地收集水果。然而,农场的主人设定了一些严格的规矩,你必须按照要求采摘水果:
- 你只有 两个 篮子,并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
子数组中的 数字 只能有两种
- 你可以选择任意一棵树开始采摘,你必须从 每棵 树(包括开始采摘的树)上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次,你将会向右移动到下一棵树,并继续采摘。
子数组是连续的
- 一旦你走到某棵树前,但水果不符合篮子的水果类型,那么就必须停止采摘。
给你一个整数数组 fruits ,返回你可以收集的水果的 最大 数目。
也即返回 最长的,连续的,数字 只能有两种的 子数组
示例 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] 这五棵树。
本题和 LeetCode209 相似,同样使用
滑动窗口
即可解决;
思路:
- 使用map存储数字种类
- 一旦超过两种,开始将起始位置的左移,但是最初本人思维受限于一种情况也即
3,3,3,1,2,1,1,2,3,3,4
,一直在思考如何将 前三个3直接从map中去掉,但是这种想法是错的, - 当出现
3,1,3,1,2....
这种情况,就不能确定 删map里的 3还是1,因此是需要把他们作为key逐个减少他们的个数, - 当某个key的个数为0时,即可真正从map中删除
class Solution {
public int totalFruit(int[] fruits) {
HashMap<Integer,Integer> map = new HashMap<>();
int left = 0;
int max = 0;
for (int i = 0; i < fruits.length; i++) {
map.put(fruits[i],map.getOrDefault(fruits[i],0)+1);
//装的水果种类超过两个时
while(map.size()>2){
//以最初遍历的顺序开始抛弃水果,相当于左指针开始右移
map.put(fruits[left],map.get(fruits[left])-1);
//当 某个篮子的水果抛弃完后,则说明当前只剩一种水果,可以添加下一种水果
if(map.get(fruits[left])==0){
//完全抛弃这一种水果
map.remove(fruits[left]);
}
//真正的左指针右移,滑动窗口的起始位置开始移动
left++;
}
max = Math.max(max,i-left+1);
}
return max;
}
}