持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第32天，点击查看活动详情
你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 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 <= 10^50 <= fruits[i] < fruits.length
思路
本题可以用滑动窗口求解。left和right分别为滑动窗口的左右边距，只需要确保滑动窗口中只有两种水果，left和right的最大间距就是我们要求的解。定义一个map用来存放当前滑动窗口中每种水果的数量，每次把right向前移动一位，如果map中已存在fruits[right]，把数量+1，如果不存在，把fruits[right]放入map中，并设置数量为1。如果map中水果数量大于两种，把left向前移动一位，并把map中fruits[left]数量-1，如果数量为0，移出map，重复上述步骤，只到map中水果为两种。
解题
/**
* @param {number[]} fruits
* @return {number}
*/
var totalFruit = function (fruits) {
let left = 0;
let res = 0;
const map = new Map();
for (let i = 0; i < fruits.length; i++) {
let count = map.get(fruits[i]) || 0;
map.set(fruits[i], count + 1);
while (map.size > 2) {
res = Math.max(res, i - left );
count = map.get(fruits[left]);
if (count === 1) {
map.delete(fruits[left]);
} else {
map.set(fruits[left], count - 1);
}
left++;
}
}
res = Math.max(res, fruits.length - left);
return res;
};
