携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第16天，点击查看活动详情 >>

水果成篮

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

解题思路:

滑动窗口法解答，本质是求元素不超过两个的最大字串长度，使用count进行N个元素的记录，map记录每个元素出现几次
右边界进行移动，将元素根据是否在原来map对象存储过进行不同存储操作，同时注意count数值改动，如果count <= 2，则进行长度的记录，如果count > 2，此时需要移动左边的指针，压缩窗口，丢弃所有元素，直到只存在两个元素，记录长度

我的答案:

/**
 * @param {number[]} fruits
 * @return {number}
 */
var totalFruit = function(fruits) {
    let left = 0 
    let map = {}
    let count = 0
    let res = 0
    for(let i = 0;i<fruits.length;i++){
        // 新水果
        if(!map[fruits[i]]){
            count ++
            map[fruits[i]] = 1  
        } else{
            // 旧水果
            map[fruits[i]] ++
        }
        if(count<=2){
            res = Math.max(res,i-left+1)
        } else{
            // 超过两种
            while(count>2){
                map[fruits[left]] --
                if( map[fruits[left]] ===0){
                    count --
                }
                left ++
            }
        }
    }
    return res
};

最后

如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´･ᴗ･` )

LeetCode Everyday - 水果成篮