leetcode--904.水果成篮

题目描述

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

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

我的

我的想法是两个篮子，那我就把篮子给确定下来，然后把每个篮子的水果数量记录一下，如果出现第三种水果，start指针开始移动，直到某一种水果的数量为0，那就说明有个篮子空了，那我们第三种水果就可以放了。

class Solution {
    public static int totalFruit(int[] fruits) {
        int start = 0;
        int end = 0;
        int ans = 0;
        int s1 = fruits[start];
        int s2 = fruits[start];
        int count_1 = 0;
        int count_2 = 0;
        while (end < fruits.length) {
            int s3 = fruits[end];
            while (count_1 > 0 && count_2 >0
                    && s3 != s1 && s3 != s2) {
                ans = Math.max(ans, end - start);
                if (s1 == fruits[start]){
                    count_1 --;
                }else if (s2 == fruits[start]) {
                    count_2 --;
                }
                start++;
            }
            if (count_1 == 0) {
                s1 = s3;
                count_1 = 1;
            }else if (count_1 != 0 && s3 == s1) {
                count_1 ++;
            }else if (count_2 == 0) {
                s2 = s3;
                count_2 = 1;
            }else if (count_2 != 0 && s3 == s2) {
                count_2 ++;
            }
            end++;
        }

        if (end - start > ans) {
            ans = end - start;
        }

        return ans;

    }
}

答案

class Solution {
    public int totalFruit(int[] tree) {
        int ans = 0, i = 0;
        Counter count = new Counter();
        for (int j = 0; j < tree.length; ++j) {
            count.add(tree[j], 1);
            while (count.size() >= 3) {
                count.add(tree[i], -1);
                if (count.get(tree[i]) == 0)
                    count.remove(tree[i]);
                i++;
            }

            ans = Math.max(ans, j - i + 1);
        }

        return ans;
    }
}

class Counter extends HashMap<Integer, Integer> {
    public int get(int k) {
        return containsKey(k) ? super.get(k) : 0;
    }

    public void add(int k, int v) {
        put(k, get(k) + v);
    }
}

作者：LeetCode
链接：https://leetcode-cn.com/problems/fruit-into-baskets/solution/shui-guo-cheng-lan-by-leetcode/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

总结

答案是用了counter类来进行"+“和”-"的统一，本质上和我的思路差不多，还是要用水果类型和水果数量来进行判断，从而来判断第三种类型的水果的加入，代码量减少了很多，但是可能是引入了hashmap，导致执行时间和耗存都比较大。

最下面是我自己写的，上面两个是hashmap。