LeetCode 每日一题：水果成篮

持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第 7 天，点击查看活动详情。

水果成篮

原题地址

你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 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^5$
• 0 <= fruits[i] < fruits.length

思路分析

1. 分析题目，可以得知题目最终需要得到的是一个数组间的元素差值不超过2的最长子数组，可以使用滑动窗口来解决该问题；
2. 那么定义 left 和 right 分别表示满足条件的左右边界，定义 count 以数为 key，次数为 value，来存储这个区间内的数；
3. 以 left 为定值来移动寻找 right 的值，right 每移动一次就将 fruits[right] 存储，如果 count 中超过两个键值对，则表示已经找到当前 left 下的最远 right 值，然后再移动 left 寻找下一个满足条件的 right 值；
4. 在遍历移动过程中，使用 res 值来存储最大的区间值，最后得到的 res 值即为满足条件的结果。

AC 代码

/**
 * @param {number[]} fruits
 * @return {number}
 */
var totalFruit = function(fruits) {
    const count = new Map()
    let left = 0, res = 0
    for (let right = 0; right < fruits.length; right++) {
        count.set(fruits[right], (count.get(fruits[right]) || 0) + 1)
        while (count.size > 2) {
            count.set(fruits[left], count.get(fruits[left]) - 1)
            if (count.get(fruits[left]) === 0) {
                count.delete(fruits[left])
            }
            left++
        }
        res = Math.max(res, right - left + 1)
    }
    return res
};

结果：

• 执行结果： 通过
• 执行用时：120 ms, 在所有 JavaScript 提交中击败了17.43%的用户
• 内存消耗：52.7 MB, 在所有 JavaScript 提交中击败了35.52%的用户
• 通过测试用例：91 / 91

END