持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第21天，点击查看活动详情
你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 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] 这五棵树。
提示：
1 <= fruits.length <= 1050 <= fruits[i] < fruits.length
解题思路: (滑动窗口)
本题我们可以使用滑动窗口来解决
滑动窗口: 一个可以在数组中来回移动的窗口，通过不断的移动窗口的起始位置和结束位置，来找出最终结果，属于双指针的一种
我们就需要使用滑动窗口来找出最终的结果，还要一个数组nums来记录现在窗口中两种水果出现的次数，还要一个count来记录窗口里的水果种类
- 如果
count <= 2,rigth就可以一直向右移 - 当
count > 2时，表示窗口内水果种类已经超过2种，为了减少窗口内的水果种类，我们只能将左指针left不断的向右移动，直到窗口内的水果种类减少到2种，这里要注意的是，每次移动需要更新数组nums - 最后取最大值即可
代码: (Java实现)
public int totalFruit(int[] fruits) {
int len = fruits.length;
if (len <= 2) {
return len;
}
//左指针
int left = 0;
//右指针
int right = 0;
//窗口内水果种类数
int count = 0;
//默认的最大值，如果数组里只有两个元素，返回这个数
int res = 2;
//记录现在窗口内中的每种水果的出现次数
//因为提示`0 <= fruits[i] < fruits.length`说明了水果种类是有限的
int[] nums = new int[len];
while (right < len) {
nums[fruits[right]] += 1;
if (nums[fruits[right]] == 1) {
count++;
}
right++;
while (count > 2) {
nums[fruits[left]] -= 1;
if (nums[fruits[left]] == 0) {
count--;
}
left++;
}
//不断取最大值
res = Math.max(res,right-left);
}
return res;
}
复杂度分析:
- 时间复杂度: O(n)
- 空间复杂度: O(1)
