持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第15天，点击查看活动详情

1. 题目与解析

你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示，其中 fruits[i] 是第 i 棵树上的水果种类。

你想要尽可能多地收集水果。然而，农场的主人设定了一些严格的规矩，你必须按照要求采摘水果：

你只有两个篮子，并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。

你可以选择任意一棵树开始采摘，你必须从每棵树（包括开始采摘的树）上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次，你将会向右移动到下一棵树，并继续采摘。

一旦你走到某棵树前，但水果不符合篮子的水果类型，那么就必须停止采摘。

给你一个整数数组 fruits ，返回你可以收集的水果的最大数目。

输入： fruits = [1,2,1]

输出： 3

解释：可以采摘全部 3 棵树。

输入： fruits = [0,1,2,2]

输出： 3

解释：可以采摘 [1,2,2] 这三棵树。如果从第一棵树开始采摘，则只能采摘 [0,1] 这两棵树。

输入： fruits = [1,2,3,2,2]

输出： 4

解释：可以采摘 [2,3,2,2] 这四棵树。如果从第一棵树开始采摘，则只能采摘 [1,2] 这两棵树。

输入： fruits = [3,3,3,1,2,1,1,2,3,3,4]

输出： 5

解释：可以采摘 [1,2,1,1,2] 这五棵树。

根据题意，可以摘取的树一定是连续的，也就是说，我们需要做的是找到符合题意的最长联系子数组，因此，我们可以考虑使用双指针的当时通过一次遍历来解决问题。

2. 题解

class Solution {
    public int totalFruit(int[] fruits) {
        int tmp = 0, ans = 0, ind = 0;
        Map<Integer, Integer> map = new HashMap<>();
        for (int fruit: fruits) {
            if (map.size() >= 2 && !map.containsKey(fruit)) {
                int delFruit = fruits[ind];
                while (true) {
                    map.put(delFruit, map.getOrDefault(delFruit, 0) - 1);
                    tmp--;
                    ind++;
                    if (map.get(delFruit) == 0) {
                        map.remove(delFruit);
                        break;
                    }                  
                    delFruit = fruits[ind];
                }
            }

            map.put(fruit, map.getOrDefault(fruit, 0) + 1);
            tmp++;
            ans = Math.max(ans, tmp);
        }
        return ans;
    }
}

leetcode 904. 水果成篮

1. 题目与解析

2. 题解