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

力扣——904. 水果成篮

904. 水果成篮 - 力扣（LeetCode）

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

提示：

• 1 <= fruits.length <= 105
• 0 <= fruits[i] < fruits.length

问题解析

双指针模拟滑动窗口。

用一个哈希表记录每个水果的出现次数，对于位于当前滑动窗口右边界的水果，我们判断一下当前水果是否之前出现过:

如果出现过则直接放入篮子

如果没出现过，则看是否当前两个篮子都用了，如果没有，则篮子消耗数++；如果都用了，记录下当前滑动窗口的长度，这就是从左边界开始可以摘的最多水果数，然后开始收缩左边界，直到消耗篮子数变回1为止。

在此过程中维护滑动窗口的最大长度，就是答案。

AC代码

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        unordered_map<int,int>mymap;
        int n=fruits.size(),l=0,r=0,mx=0;
        while(r<n)
        {
            mymap[fruits[r]]++;
            r++;
            if(mymap.size()<=2)
            {
                int sum=0;
                for(auto i:mymap)
                {
                    sum+=i.second;
                }
                mx=max(mx,sum);
            }
            while(mymap.size()>2)
            {
                mymap[fruits[l]]--;
                if(mymap[fruits[l]]==0)
                    mymap.erase(fruits[l]);
                l++;
            }
        }
        if(mymap.size()<=2)
        {
            int sum=0;
            for(auto i:mymap)
            {
                sum+=i.second;
            }
            mx=max(mx,sum);
        }
        return mx;
    }
};