【LeetCode-904】水果成篮

问题

在一排树中，第 i 棵树产生 tree[i] 型的水果。
你可以从你选择的任何树开始，然后重复执行以下步骤：

把这棵树上的水果放进你的篮子里。如果你做不到，就停下来。

1. 移动到当前树右侧的下一棵树。如果右边没有树，就停下来。
2. 请注意，在选择一颗树后，你没有任何选择：你必须执行步骤 1，然后执行步骤 2，然后返回步骤 1，然后执行步骤 2，依此类推，直至停止。

你有两个篮子，每个篮子可以携带任何数量的水果，但你希望每个篮子只携带一种类型的水果。

用这个程序你能收集的水果树的最大总量是多少？

示例

输入： [3,3,3,1,2,1,1,2,3,3,4]
输出： 5
解释： 我们可以收集 [1,2,1,1,2]

解答

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        int l = 0, r = 0, res = 0, cnt = 0, n = fruits.size();
        int memo[n]; bzero(memo, sizeof memo);
        while (r < n) {
            if (memo[fruits[r++]]++ == 0) cnt++;
            while (cnt > 2)
                if (memo[fruits[l++]]-- == 1) cnt--;
            res = max(res, r - l);
        }
        return res;
    }
};

重点思路

本题需要求解不超过2种数字的最长子数组。经典的atMost问题。本题需要逐步收缩左窗口，直到满足条件才停下，而非类似【LeetCode-795】区间子数组个数，当不满足条件时，左指针直接跳转到右指针位置。