在一排树中,第 i
棵树产生 tree[i]
型的水果。
你可以从你选择的任何树开始,然后重复执行以下步骤:
- 把这棵树上的水果放进你的篮子里。如果你做不到,就停下来。
- 移动到当前树右侧的下一棵树。如果右边没有树,就停下来。
请注意,在选择一颗树后,你没有任何选择:你必须执行步骤 1,然后执行步骤 2,然后返回步骤 1,然后执行步骤 2,依此类推,直至停止。
你有两个篮子,每个篮子可以携带任何数量的水果,但你希望每个篮子只携带一种类型的水果。
用这个程序你能收集的水果总量是多少?
示例 1:
输入:[1,2,1] 输出:3 解释:我们可以收集 [1,2,1]。
示例 2:
输入:[0,1,2,2] 输出:3 解释:我们可以收集 [1,2,2]. 如果我们从第一棵树开始,我们将只能收集到 [0, 1]。
示例 3:
输入:[1,2,3,2,2] 输出:4 解释:我们可以收集 [2,3,2,2]. 如果我们从第一棵树开始,我们将只能收集到 [1, 2]。
示例 4:
输入:[3,3,3,1,2,1,1,2,3,3,4] 输出:5 解释:我们可以收集 [1,2,1,1,2]. 如果我们从第一棵树或第八棵树开始,我们将只能收集到 4 个水果。
解答如下:
class Solution {
public:
int totalFruit(vector<int>& tree) {
int size=tree.size();
if(size<1)
return 0;
if(size<=2)
return size;
//模拟队列,其中que[1]表示的是当前的结尾元素,que[0]表示的是在前面的元素
vector<int> que(2);
que[0]=tree[0];
int i=1;
int maxa=1;//已经放入1个
//此for循环,是为了处理形如0,0,0,1类似的tree
for(;i<size;i++){
if(tree[i]==que[0]){
maxa++;
}else{
que[1]=tree[i];
maxa++;
i++;
break;
}
}
int result=0;
int len=1;//如10000006,则对最后一个0,处理后len=6,当迎来新的6时,maxa=6+1
for(;i<size;i++){
if(tree[i]==que[0]){
//当tree[i]和前面的元素一致时,这是tree[i]就应当做结尾,故交换
swap(que[0],que[1]);
len=1;//舍弃以之前队尾统计的结果
maxa++;
continue;
}
if(tree[i]==que[1]){
len++;//与结尾相同+1
maxa++;
continue;
}
//既不等于que[1],又不等于que[0]
//比较上一次maxa与result,取最大值
result=max(result,maxa);
//初始化长度
maxa=len+1;
len=1;
//出队摒弃旧元素,入队迎接新元素
que[0]=que[1];
que[1]=tree[i];
}
result=max(result,maxa);
return result;
}
};