904. 水果成篮
题意:
在一排树中,第 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 棵水果树。
解题思路:
这道题的其实可以理解为两个数字最大的子数组.我用unordered_map容器,遍历数组压容器,如果超过两个数字,那么容器的长度就为3,这样就可以计算前两个数字的长度,然后类推下去找出最大的长度即可.
代码:
#include<iostream>
using namespace std;
#include<vector>
#include<unordered_map>
int main()
{
vector<int>fruits = { 0,1,2,2 };
if (fruits.size() == 0)
return 0;
if (fruits.size() == 1)
return 1;
unordered_map<int, int>nums;
int left = 0;
int len = 0;
int maxlen = 0;
for (int right = 0; right < fruits.size(); right++)
{
nums[fruits[right]] = nums[fruits[right]] + 1;
if (nums.size() > 2)
{
len = nums.begin()->second + (++nums.begin())->second;
nums.erase(nums.begin());
}
if (maxlen < len)
maxlen = len;
}
len = nums.begin()->second + (++nums.begin())->second;
if (maxlen < len)
maxlen = len;
cout << maxlen;
return 0;
}
vs运行结果
问题:
我在VS2017是实现后,运行结果没问题.但是到了leetcode编译器上面就出现问题了,我也不知道啥原因.如果有大佬路过可以帮我看一下为什么,代码是一样的搬运过去leetcode而已
leetcode运行结果
总结:
我感觉这样的想法应该没问题,但是在leetcode上不知道错误在哪里,又没有开会员,所以没办法用leetcode上的调试器.所以希望有大佬可以给点提示.