Leetcode-904-水果成篮
你正在探访一家农场,农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示,其中 fruits[i] 是第 i 棵树上的水果 种类 。
你想要尽可能多地收集水果。然而,农场的主人设定了一些严格的规矩,你必须按照要求采摘水果:
·你只有 两个 篮子,并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
·你可以选择任意一棵树开始采摘,你必须从 每棵 树(包括开始采摘的树)上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。
每采摘一次,你将会向右移动到下一棵树,并继续采摘
·一旦你走到某棵树前,但水果不符合篮子的水果类型,那么就必须停止采摘。
示例:
输入:fruits = [1,2,1]
输出:3
解释:可以采摘全部 3 棵树。
解法一:滑动窗口
class Solution {
public:
int totalFruit(vector<int>& fruits) {
int typeA,typeB;
int i,j,max_length,length;
length=fruits.size();
typeA=fruits[0];
j=0;
i=0;
max_length=0;
while(j<length&&fruits[j]==typeA) j++;
if(j==length)
return j;
typeB=fruits[j];
for(;j<fruits.size();j++){
if(fruits[j]!=typeA&&fruits[j]!=typeB){
max_length=max(max_length,j-i);
typeA=fruits[j-1];
typeB=fruits[j];
i=j-1;
while(fruits[i-1]==typeA)
i--;
}
}
max_length=max(j-i,max_length);
return max_length;
}
};
踩坑点:
例如:while(fruits[j]==typeA&&j<length) j++;
如果j<length条件放在前面,若输入[0]
,此时j=1,将fruit[1]和typeA比较会导致溢出,因此正确的做法是while(j<length&&fruits[j]==typeA)
,将j<length放在前面,由于逻辑与的短路效应,在第一个条件错误后,直接跳过第二个条件。
解法二:哈希表+滑动窗口(官方解法)
class Solution {
public:
int totalFruit(vector<int>& fruits) {
int left,right,length,result;
unordered_map<int,int> hash_map;
length=fruits.size();
left=right=0;
for(;right<length;right++){
++hash_map[fruits[right]];
while(hash_map.size()>2){
auto it=hash_map.find(fruits[left]);
it->second--;
if(it->second==0){
hash_map.erase(it);
}
left++;
}
result=max(result,right-left+1);
}
return max(result,right-left);
}
};
哈希表记录的是当前子窗口两种类型的果树各自的数量,当数量为0时将其抹除,且哈希表必须满足length<=2,当大于2时说明有第三种果树,此时需要将与第三种果树不相邻
的类型抹除,因此left往前走,直到某一类型完全为0,此时还剩下新加入的第三种类型和紧挨着的另一种类型,例如[1,2,1,2,2,2,3]
,当3加入时,left会移动到index=3处时停下来,因为此时hash_map的长度为2,开启新一轮的计数