力扣第904题

一、题目：904. 水果成篮

二、题目解析：

题目解析：题目不太好理解，通俗解释一下，可以把数组中的每个元素理解成一个树，元素值理解成那种类型的果树，比如:[3,3,3,1,2,1,1,2,3,3,4]
我们可以认为1是一个苹果树，2是一个梨树，3是一个橘子树，4是一个枣树，所以这里上述数组里面有四种类型的果树。同时，我们有两个篮子，我们需要用这两个篮子装这些树的水果，但是有一个条件，每个篮子只能装相同的水果，注意改题的限制条件–>只能装连着的水果，中间不能断，不能插入其他类型的水果。
改题可以借鉴一下力扣第三题。属于是滑动窗口的题目
解题步骤：

1. 创建一个篮子Map,这里Map的key代表的是水果的种类，value是这种水果在数组中的位置
2. 遍历数组，往Map中添加水果并更新对应小标
3. 当Map里面的水果超过2种，我们就选择下标比较小的水果进行移除，同时移除一种水果后需要更新 j 的位置
4. 记录并更新篮子中水果个数最大值

图示帮助理解：

三、代码如下：

public int totalFruit(int[] fruits) {
        //创建一个篮子，这里用map做了一定优化
        //当篮子里放两种水果的时候我们真正关心的是从前面移除一个一个的水果，直到把其中一种水果完全的从篮子里移除掉，就是放第三种水果的时候
        //所以我们真正关心的是每一个水果它最后出现的位置，这里Map的key代表的是水果的种类，value是这种水果在数组中的位置
        //引入Map简化了滑动窗口算法一点一点关窗动作，直接找到要关窗的位置
        Map<Integer,Integer> map = new HashMap<>();
        //边界值 1 <= fruits.length <= 10^5
        int maxLength = 1;
        int j = 0;
        for(int i = 0; i<fruits.length; i++){
            //添加水果并更新对应小标
            map.put(fruits[i],i);
            //当Map里面的水果超过2种，我们就选择下标比较小的水果进行移除
            if(map.size()>2){
                int minIndex = fruits.length-1;
                //这里for循环实际只有两个数，这里需要找到较小水果种类的下标
                for (Integer index: map.values()) {
                    if( index < minIndex ){
                        minIndex = index;
                    }
                }
                //删除数组中靠前的水果
                map.remove(fruits[minIndex]);
                //移除一种水果后需要更新 j 的位置
                j = minIndex + 1;
            }
            // 篮子中水果个数 即 i - j + 1
            maxLength = Math.max(maxLength, i - j + 1);
        }
        return maxLength;
}

四、测试

五、结束