力扣之水果成篮史上最全解法

Posted 2022-03-30 super码王

题目连接

904. 水果成篮 - 力扣（LeetCode） (leetcode-cn.com)https://leetcode-cn.com/problems/fruit-into-baskets/

题目描述

解题思路

像这种求最大长度的问题，滑动窗口成了最热门的解法。

所谓 滑动窗口 ，在我看来其实就是左右两个指针，右指针负责遍历数据，左指针根据条件移动，目标长度 = 右指针 - 左指针 + 1

有了上面的大局观，我们就来结合本题目好好用一下滑动窗口：

1. 定义一个右指针 right，使用它来遍历数据列表。
2. 定义一个分类数组，每遍历一个元素的时候就比对分类数组中是否存在该元素，如果不存在，则根据分类数组中元素个数（种类）来决定是直接添加元素还是更新分类数组
3. 定义一个类型变化的指针，这个指针的主要用途是方便 步骤2 在更新分类数组的时候 赋值给左指针。
4. 计算最大长度即可

代码实现

let dataList = [1, 0, 1, 4, 1, 4, 1, 2, 3]

var totalFruit = function (fruits) 

    let left = 0
    let right
    let classArray = []
    let n = 0
    let maxLen = 0

    //定义一个右指针，遍历整个数组
    for (right = 0; right < fruits.length; right++) 

        //如果发现新的水果类型，进入此代码块
        if (!classArray.includes(fruits[right])) 

            if (classArray.length <= 1) 
                //如果水果类型小于等于1，直接往里面加水果类型
                classArray.push(fruits[right])
             else 
                //如果水果类型大于1，更新分类数组
                classArray[0] = fruits[right]
                classArray[1] = fruits[right - 1]
                left = n
            
        

        //记录水果种类变动索引
        if (fruits[right] != fruits[right - 1]) 
            n = right
        

        //计算最大长度
        maxLen = Math.max(maxLen, right - left + 1)
    

    return maxLen



console.log(totalFruit(dataList))

运行结果