904. 水果成篮 【@我的刷题日记】

难度：中等        日期：2022年11月11日

你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 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] 这五棵树。

方法：滑动窗口

设置左指针和右指针，每当遍历了2个以上不同类型的水果以后，就将左指针滑动到与当前右指针前一个数字同一类型连续位置的第一个（如：1222345，如果右指针到了数字‘3’位置，已经遍历了2种以上不同类型的数了，因此左指针就要更新到3个2中第一个'2'的位置），记录当前左右指针之间的数字个数，与之前的值相比较取较大的那个值。

答案一：官方答案

其中Map.getOrDefault(Object key, V defaultValue)方法的作用是：当Map集合中有这个key时，就使用这个key值；如果没有就使用默认值defaultValue。

创建一个map用来记录遍历过的数字种类数（size）和遍历过的每种类型数字个数（value）。当map中种类数超过2种时，就将left前移直到map中只有两种不同的数字种类。

class Solution {
    public int totalFruit(int[] fruits) {
        int n = fruits.length;
        // 创建一个Map用来记录遍历过的数字种类数 和 遍历过的每种类型数字个数
        Map<Integer, Integer> cnt = new HashMap<Integer, Integer>();

        int left = 0, ans = 0;
        for (int right = 0; right < n; ++right) {
            // 将fruits[right]压入map，这个数在map中存在值就+1，不存在就设为1
            cnt.put(fruits[right], cnt.getOrDefault(fruits[right], 0) + 1);
            // 如果map中存在两种数字类型以上的数
            while (cnt.size() > 2) {
                // 左指针逐渐前移，直到map中只有两种不同的数字
                cnt.put(fruits[left], cnt.get(fruits[left]) - 1);
                if (cnt.get(fruits[left]) == 0) {
                    cnt.remove(fruits[left]);
                }
                ++left;
            }
            // 更新ans记录最长个数
            ans = Math.max(ans, right - left + 1);
        }
        return ans;
    }
}

结果

答案二：优质题解

思路相似，但是不同的是没有用map存储，而是用计数排序的方式记录遍历过的数字种类数和遍历过的每种类型数字个数。

class Solution {
    public int totalFruit(int[] fs) {
        int n = fs.length, ans = 0;
        int[] cnts = new int[n + 10];
        for (int i = 0, j = 0, tot = 0; i < n; i++) {
            if (++cnts[fs[i]] == 1) tot++;
            while (tot > 2) {
                if (--cnts[fs[j++]] == 0) tot--;
            }
            ans = Math.max(ans, i - j + 1);
        }
        return ans;
    }
}

结果

答案三：自己写的 

不同点在于用一个长度为2的数组存储。

class Solution {
    public int totalFruit(int[] fruits) {
        int left = 0;
        int[] tool = new int[]{fruits[0],fruits[0]};
        int sum = -1;
        for(int right = 0; right < fruits.length ; right++){
            if(fruits[right] != tool[0] && fruits[right] != tool[1] && tool[0] != tool[1]){
                left = right-1;
                while(fruits[left-1] == fruits[left]){
                    left--;
                }
                tool[0] = fruits[left];
            }
            if(fruits[left] != fruits[right]){
                // tool[0] = fruits[left];
                tool[1] = fruits[right];
            }
            sum = sum > right - left + 1?sum : right - left + 1;
        }

        return sum;

    }
}