leetcode 水果成篮

水果成篮

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

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

你只有 两个 篮子，并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
你可以选择任意一棵树开始采摘，你必须从 每棵 树（包括开始采摘的树）上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次，你将会向右移动到下一棵树，并继续采摘。
一旦你走到某棵树前，但水果不符合篮子的水果类型，那么就必须停止采摘。
给你一个整数数组 fruits ，返回你可以收集的水果的 最大 数目。

限制
1 <= fruits.length <= 100000
0 <= fruits[i] < fruits.length

示例

示例 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] 这五棵树。

具体含义

树上一定有果子，只是种类不同。
你一定得摘果子，每次遇到新种类还得摘一种新的。
篮子最多放两种，数量不限。

例如例二，0代表树上果子的种类不在重复列表中，摘了之后单独占一个格子，故01之后放不了2或者3之类的。
如例三，1摘了之后占一个格子，2再占一个，这个时候3出现，为了摘更多种类的，放不下了，从1开始就最多2棵树。从2开始，占一个，到3这时摘跟2重复的交集的那种就行，后续的2就仍在果篮的种类范围内，便可继续摘。

或者换句话说，查找包含元素种类不超过两种的最长子串，返回子串长度。

思路

滑动窗口+哈希

思路与力扣76. 最小覆盖子串几乎相同，只是那个题操作的是字符串。

左指针和右指针起始位置相同，右指针移动，种类表（哈希表，也就是篮子）记录该类型的出现次数，同时出现次数满一次就记类型出现了。
当超过两种就放不下了，此时移动左指针，移动就从篮子里取走，种类表对应减一。种类表对应为0时就取光了，种类计数减一，使得跳出循环。
这时计算滑动窗口长度并与最大值进行比较和保存，完成后再移动右指针，继续循环。

时间复杂度为 O(n)，其中 n 是 fruits 数组的长度，空间复杂度为 O(1)，因为 type 数组的大小是固定的，不随 fruits 数组的大小变化。
（leetcode测试例子有数很大的所以type很大，参见限制）

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        int left = 0, right = 0, maxLen = 0;
        int type[100001] = {0}; // 假设水果编号从 1 到 100001 题目数字大小限制
        int count = 0; // 记录当前窗口中不同水果的类型数

        while (right < fruits.size()) {
            type[fruits[right]]++; // 增加当前水果的计数
            if (type[fruits[right]] == 1) { // 如果是新的水果类型
                count++; // 增加水果类型计数
            }

            // 当有超过两种类型的水果时，移动 left 指针
            while (count > 2) {
                type[fruits[left]]--;
                if (type[fruits[left]] == 0) { // 如果类型计数为 0，说明这种水果已经全部移出窗口
                    count--; // 减少水果类型计数
                }
                left++; // 移动 left 指针向前
            }

            // 更新最长长度
            maxLen = max(maxLen, right - left + 1);
            right++; // 移动 right 指针向前
        }

        return maxLen;
    }
};