水果成篮(Java)

（leetcode 904）

题目要求

你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示，其中 fruits[i] 是第 i 棵树上的水果 种类 。
你想要尽可能多地收集水果。然而，农场的主人设定了一些严格的规矩，你必须按照要求采摘水果：

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

思路

滑动窗口。
用集合HashMap存储水果和数量。
HashMap的大小不能大于2。

HashMap

存储单元是一个键值对，即key-value

常用方法

添加、删除、修改
Object put(Object key,Object value)//将指定key-value添加到(或修改)当前map对象中
void putAll(Map m)//将m中的所有key-value对存放到当前map中
Object remove(Object key)//移除指定key的key-value对，并返回value
void clear()//清空当前map中的所有数据
元素查询
Object get(Object key)//获取指定key对应的value
Object getOrDefault(Object key, V defaultValue)//获取value，找不到返回defaultValue
boolean containsKey(Object key)//是否包含指定的key
boolean containsValue(Object value)//是否包含指定的value
int size()//返回map中key-value对的个数
boolean isEmpty()//判断当前map是否为空
boolean equals(Object obj)//判断当前map和参数对象obj是否相等

代码

public int totalFruit(int[] fruits) {
        //HashMap key为水果种类，value为水果数量
        Map<Integer,Integer> basket = new HashMap<>();

        int result = 0;
        int l = 0;

        //右边界滑动
        for(int r = 0; r < fruits.length; r ++) {
            //加入篮子
            basket.put(fruits[r], basket.getOrDefault(fruits[r], 0) + 1);

            //左边界滑动
            while(basket.size() > 2) {
                basket.put(fruits[l], basket.get(fruits[l]) - 1);
                if(basket.get(fruits[l]) == 0) basket.remove(fruits[l]);
                l ++;
            }
            result = Math.max(result, r - l + 1);
        }
        return result;
    }