LeeCode 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

 

没啥说的，使用HashMap实现滑动窗口，代码如下：

    public int totalFruit(int[] tree) {
        int head=0;
        int tail=0;
        int max = 0;
        int len = tree.length;
        Map<Integer,Integer> map = new HashMap<>();
        while(head<len&&tail<len){
            if(!map.containsKey(tree[tail])){
                if(map.size()<2){
                    max = tail-head+1>max?tail-head+1:max;
                    map.put(tree[tail],1);
                    tail++;
                }else{
                    if(map.get(tree[head])>1){
                        map.put(tree[head],map.get(tree[head])-1);
                        head++;
                    }else{
                        map.remove(tree[head]);
                        head++;
                    }
                }
            }else{
                max = tail-head+1>max?tail-head+1:max;
                map.put(tree[tail],map.get(tree[tail])+1);
                tail++;
            }
        }
        return max;
    }