每天一道算法题之水果成篮

题目描述:

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

1. 把这棵树上的水果放进你的篮子里。如果你做不到，就停下来。
2. 移动到当前树右侧的下一棵树。如果右边没有树，就停下来。

请注意，在选择一颗树后，你没有任何选择：你必须执行步骤 1，然后执行步骤 2，然后返回步骤 1，然后执行步骤 2，依此类推，直至停止。

你有两个篮子，每个篮子可以携带任何数量的水果，但你希望每个篮子只携带一种类型的水果。
用这个程序你能收集的水果总量是多少？¹

示例 1：

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

• 分析: 滑动窗口算法
  • 窗口中只能出现两种类型的水果,对水果数量没有要求,采用哈希表存储窗口中的数据,当哈希表的大小等于3时,减少窗口,
• 代码:

class Solution {
    public int totalFruit(int[] tree) {
        //滑动窗口算法
        //采用哈希集表录窗口中的数据:窗口中存在的水果类型
        int left = 0,right = 0;
        int len = tree.length;
        HashMap<Integer,Integer> map = new HashMap<>();
        int ans = 0;
        
        while(right<len){
            map.put(tree[right],map.getOrDefault(tree[right],0)+1);
            right++;

            while(left < tree.length && map.size()==3){
                int lNum = map.getOrDefault(tree[left],0);
                ans = Math.max(ans,right -1 - left);
                if(lNum==1){
                    map.remove(tree[left]);
                }else{
                    map.put(tree[left],lNum-1);
                }

                left++;
            }
        }
        //比较最后一次的窗口,
        ans = Math.max(ans,right - left);
        return ans;
    }
}

---

1. https://leetcode-cn.com/problems/fruit-into-baskets/submissions/ ↩︎