题目: 你正在探访一家农场,农场从左到右种植了一排果树。这些树用一个整数数组 fruits
表示,其中 fruits[i]
是第 i
棵树上的水果 种类 。你想要尽可能多地收集水果。然而,农场的主人设定了一些严格的规矩,你必须按照要求采摘水果:
- 你只有 两个 篮子,并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
- 你可以选择任意一棵树开始采摘,你必须从 每棵 树(包括开始采摘的树)上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次,你将会向右移动到下一棵树,并继续采摘。
- 一旦你走到某棵树前,但水果不符合篮子的水果类型,那么就必须停止采摘。
给你一个整数数组 fruits
,返回你可以收集的水果的 最大 数目。
惯例: 献上我的笨蛋解法,虽然我知道他是对的,但他遇到超级多果树测试的时候,超时了,我甚至不甘心,有放到了vs上面跑,结果软件崩了。经供娱乐,不能参考。
class Solution {
public:
int totalFruit(vector<int>& fruits) {
int n = fruits.size();
int sum = 0;
int left = 0;
int x1 = -1;
int x2 = -1;
int max = 0;
for(int right = 0;right < n;right++){
if(fruits[right] != x1 && x1 == -1){
x1 = fruits[right];
sum++;
max = sum > max? sum:max;
}
else if(fruits[right] != x1 && fruits[right] != x2 && x2 == -1){
x2 = fruits[right];
sum++;
max = sum > max? sum:max;
}
else if(fruits[right] == x1 || fruits[right] == x2){
sum++;
max = sum > max? sum:max;
}
else if(fruits[right] != x1 && fruits[right] != x2){
max = sum > max? sum:max;
sum = 0;
x1 = -1;
x2 = -1;
left++;
right = left - 1;
}
}
return max;
}
};
①滑动窗口
class Solution {
public:
int totalFruit(vector<int>& fruits) {
unordered_map<int, int> fruitCount; // 记录当前窗口中水果类型的数量
int left = 0; // 窗口的左边界
int maxSum = 0; // 窗口的最大长度
for (int right = 0; right < fruits.size(); right++) {
fruitCount[fruits[right]]++; // 更新当前水果类型的数量
// 如果当前窗口中的水果类型数量超过了两种,则需要调整窗口的左边界
while (fruitCount.size() > 2) {
fruitCount[fruits[left]]--; // 减少左边界水果类型的数量
if (fruitCount[fruits[left]] == 0) {
fruitCount.erase(fruits[left]); // 如果数量减为0,则移除水果类型
}
left++; // 调整左边界
}
maxSum = max(maxSum, right - left + 1); // 更新窗口的最大长度
}
return maxSum;
}
};
②排行第一的
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
vector<vector<int>> m; // 用于记录水果的位置信息
int totalFruit(vector<int>& fruits) {
int res = 0, n = fruits.size(); // 初始化结果和水果数组的长度
int x = 0, y = -1; // 初始化两个水果篮子的位置,默认-1表示未找到第二种水果
// 初始化水果位置信息
m.assign(n, vector<int>());
m[fruits[0]].push_back(0);
vector<int> lastIndices(n,0); // 记录每种水果上次出现的位置
int t = 0; // 记录当前窗口的起始位置
for (int i = 1; i < n; ++i) {
// 如果第二种水果篮子未找到且当前水果不等于第一种水果
if (y == -1 && fruits[i] != fruits[x]) y = i;
// 如果当前水果不是第一种水果且不是第二种水果
if (fruits[i] != fruits[x] && fruits[i] != fruits[y]) {
// 更新结果,计算当前窗口的长度并更新起始位置
res = max(res, i - t);
t = min(lastIndices[fruits[x]], lastIndices[fruits[y]]) + 1;
x = i - 1; // 更新第一种水果篮子的位置为上一个水果的位置
y = i; // 更新第二种水果篮子的位置为当前水果的位置
}
lastIndices[fruits[i]] = i; // 更新当前水果的位置信息
}
// 更新结果,考虑最后一个窗口的长度
res = max(res, n - t);
return res;
}
};
谢谢大家,继续努力。