一、题目
你正在探访一家农场,农场从左到右种植了一排果树。这些树用一个整数数组 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] 这五棵树。
二、思路解析
还记得我在前面的题目解析中说的吗?
一段连续区间,一眼滑动窗口。
而题目既然要求种类,那我们就可以用数组模拟一个哈希表,结合一个 kinds 变量,用于记录每个数字是否已经采集过。
在进窗口的时候,如果这个元素的值为零,这代表他是第一个进入我们数组的,所以种类 kinds 应该 +1。
而因为我们只有 2 个篮子,所以当 kinds > 2 的时候,就要出窗口。
算法实现所要注意的点基本就是这些了,具体的代码实现请看下面代码
三、完整代码
class Solution {
public int totalFruit(int[] fruits) {
int n = fruits.length;
int[] hash = new int[n+1];
int ret = 0;
int kinds = 0;
for(int left = 0 , right = 0;right < n ; right++){
int in = fruits[right];
if(hash[in] == 0){
kinds++;
}
hash[in]++;
while(kinds > 2){
int out = fruits[left];
hash[out]--;
if(hash[out] == 0){
kinds --;
}
left ++;
}
ret = Math.max(ret,right - left + 1);
}
return ret;
}
}
以上就是本篇博客的全部内容啦,如有不足之处,还请各位指出,期待能和各位一起进步!