标签：不同的   改变   描述   结束   滑动   遇到   应该   跳出循环   移动   

904.水果成篮

难度：中等

描述

在一排树中，第 i 棵树产生 tree[i] 型的水果。
你可以从你选择的任何树开始，然后重复执行以下步骤：

把这棵树上的水果放进你的篮子里。如果你做不到，就停下来。
移动到当前树右侧的下一棵树。如果右边没有树，就停下来。
请注意，在选择一颗树后，你没有任何选择：你必须执行步骤 1，然后执行步骤 2，然后返回步骤 1，然后执行步骤 2，依此类推，直至停止。

你有两个篮子，每个篮子可以携带任何数量的水果，但你希望每个篮子只携带一种类型的水果。

用这个程序你能收集的水果树的最大总量是多少？

示例 1：

输入：[1,2,1]
输出：3
解释：我们可以收集 [1,2,1]。

示例 2：

输入：[0,1,2,2]
输出：3
解释：我们可以收集 [1,2,2]
如果我们从第一棵树开始，我们将只能收集到 [0, 1]。

示例 3：

输入：[1,2,3,2,2]
输出：4
解释：我们可以收集 [2,3,2,2]
如果我们从第一棵树开始，我们将只能收集到 [1, 2]。

示例 4：

输入：[3,3,3,1,2,1,1,2,3,3,4]
输出：5
解释：我们可以收集 [1,2,1,1,2]
如果我们从第一棵树或第八棵树开始，我们将只能收集到 4 棵水果树。

提示：

1 <= tree.length <= 40000
0 <= tree[i] < tree.length

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/fruit-into-baskets
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

使用滑动窗口，双指针来解答此题，时间复杂度为O(n),空间复杂度O(1)。

代码

/**
 * @param {number[]} tree
 * @return {number}
 */
var totalFruit = function(tree) {
    let canAdd=true;  //当前能否加入果篮
    let left = 0;     //左指针
    let maxNum = 0;   //窗口最大长度（篮中最大水果数量）
    let anotherCat,idx;   //记录第二种水果的类别，idx表示遇到第三种水果left右移的位置
    // 遍历
    for(let right=0;right<tree.length;right++){
        // 如果遇到与第一种水果不同的水果
        if(tree[left]!=tree[right]){
            // 判断此时能否加入新种类水果，如果能就把当前水果
            // 设为第二种水果，记录当前idx，设置canAdd为false不能加入新种类水果
            // 如果不能加入新种类水果，则说明有第二种水果，判断当前水果是否为第二种水果
            // 如果不是，则计算窗口长度，设置canAdd为true，把左指针移到idx，right保持原
            // 位开始下一窗口
            if(canAdd==true){
                anotherCat=tree[right];
                idx = right;
                canAdd=false;
            }else{
                if(tree[right]!=anotherCat){
                    maxNum = maxNum>(right-left)?maxNum:(right-left);
                    canAdd=true;
                    left = idx;
                    right--;
                }else{
                    // 当相邻的两个水果不同种类时，改变idx
                    if(tree[right]!=tree[right-1]){
                        idx = right;
                    }
                }
            }
        }else{
            // 当相邻的两个水果不同种类时，改变idx
            // 比如[1,0,1,4,1,4,1,2,3]第一次窗口结束时，left应该变为2
            if(tree[right]!=tree[right-1]){
                idx = right;
            }
        }
    }
    // 如果maxNum在循环中没有计算过，说明一共就两种水果，直接返回数组长度
    if(maxNum==0){
        return tree.length
    }
    //跳出循环后在计算一次maxNum并与之前最大的比较 返回结果
    return maxNum = maxNum>(tree.length-left)?maxNum:(tree.length-left);
};

904.水果成篮

标签：不同的   改变   描述   结束   滑动   遇到   应该   跳出循环   移动   

原文地址：https://www.cnblogs.com/Leo2gooo/p/14824199.html