【leetcode】【2022/10/17】904. 水果成篮

问题描述：

• 你正在探访一家农场，农场从左到右种植了一排果树，这些树用一个整数数组 fruits 表示，其中 fruits[i] 是第 i 棵树上的水果种类。
• 你想要尽可能多地收集水果，然而，农场的主人设定了一些严格的规矩，你必须按照要求采摘水果：
  • 你只有两个篮子，并且每个篮子只能装单一类型的水果，每个篮子能够装的水果总量没有限制；
  • 你可以选择任意一棵树开始采摘，你必须从每棵树（包括开始采摘的树）上恰好摘一个水果；
  • 采摘的水果应当符合篮子中的水果类型，每采摘一次，你将会向右移动到下一棵树，并继续采摘；
  • 一旦你走到某棵树前，但水果不符合篮子的水果类型，那么就必须停止采摘。
• 给你一个整数数组 fruits，返回你可以收集的水果的最大数目。

核心思路：

• 该题就是要求只有两个元素的子数组的最大长度。
  • 滑动窗口模板解题即可。

代码实现：

class Solution
{
public:
    int totalFruit(vector<int>& fruits)
    {
        unordered_map<int, int> mm;
        int cnt = 0;
        int i = 0, j = 0;
        while(j < fruits.size())
        {
            ++mm[fruits[j++]];
            while(mm.size() > 2)
            {
                if(--mm[fruits[i++]] == 0)
                    mm.erase(fruits[i-1]);
            }
            cnt = max(cnt, j-i);
        }
        return cnt;
    }
};