LeetCode 刷题系列 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
 

思路：

1. 第一件事，我们把这个题目翻译成人话：

    给定一个数组 nums （ 即题目中的一排树 ），找出这样的一个子数组，满足在这个子数组内的元素值只有两种 （ 即某一段树上只有两种果实 ）。求这样的子数组的最大长度 （即两个篮子中的水果数目最大）

2. 解题思路：

     1）从第 i 个元素开始遍历数组，当发现遍历到第 j 个的时候，出现了第三种水果。

         此时，停下遍历的脚步。

      2) 计算此时子数组的长度 len = j - i +1。并记录下来。

       3) 从下标 i 开始往后移动，直到 下标 i 到下标 j 的子数组内只有两种值元素为止。

             继续步骤1)
         

class Solution {
    public int totalFruit(int[] fruits) {
  int result = 0;
        int k = 2;
        Map<Integer, Integer> map = new HashMap<>(); // 记录某个元素在当前子数组内出现的数量
        int i = 0;
        for (int j = 0; j < fruits.length; j++) {
            if (!map.containsKey(fruits[j]) || map.get(fruits[j]) == 0) { // 说明当前元素 j 没有被纳入子数组
                k--;
            }
            if(map.containsKey(fruits[j])) {
                map.put(fruits[j], map.get(fruits[j]) + 1);
            }else {
                map.put(fruits[j],1);
            }

            while (k < 0) { // 说明现在出现了第三个元素，此时，要把子数组的其中一个元素消除掉。
                // 即：把起始边界边界 i 后移到子数组某个位置，该位置往后，只有两个元素的值
                map.put(fruits[i], map.get(fruits[i]) - 1);
                if (map.get(fruits[i]) == 0) { // 下标 i+1 到 j 的子数组 中，只有两种元素值
                    k++;
                }
                i++;
            }
            result = result > (j - i + 1) ? result : (j - i + 1);
        }
        
        return result;
    }
}