力扣904. 水果成篮

力扣题目链接

1.题目介绍

题目翻译成人话就是 ：找包含两种元素的最长子串，返回其长度。

2.解法

实现滑动窗口，主要确定如下三点：
1.窗口内是什么？ 窗口是只含两种元素的子串。
2.如何移动窗口的起始位置？当窗口中出现超过两种元素（即哈希表中出现超过两个键值对），就需要不断移动 left，并将 fruits[left] 从哈希表中移除，直到哈希表满足要求为止。
3.如何移动窗口的终止位置？窗口的结束位置就是遍历整个数组的指针，也就是for循环里的索引。

法一：

class Solution {
    public int totalFruit(int[] fruits) {
        int n = fruits.length;
        Map<Integer, Integer> hashMap = new HashMap<Integer, Integer>();
        int left = 0, res = 0;
        for (int right = 0; right < n; right++) {
            //存储这个窗口内的数以及出现的次数
            hashMap.put(fruits[right], hashMap.getOrDefault(fruits[right], 0) + 1);
            //动态调节滑动窗口的起始位置
            while (hashMap.size() > 2) {
                hashMap.put(fruits[left], hashMap.get(fruits[left]) - 1);
                //出现次数减小为0时需要将对应的键值对从哈希表中移除
                if (hashMap.get(fruits[left]) == 0) {
                    hashMap.remove(fruits[left]);
                }
                left++;
            }
            res = Math.max(res, right - left + 1);
        }
        return res;
    }
}

法二：和jzh一起想出来的，可以减少时间复杂度

class Solution {
    public int totalFruit(int[] fruits) {
        Map<Integer, Integer> map = new HashMap<>();
        int n = fruits.length, ans = 0;
        for (int l = 0, r = 0; r < n; r++) {
            // map的值是记录每个键最后一次出现的位置
            if (map.containsKey(fruits[r])) {
                map.replace(fruits[r], r);
            } else {
                map.put(fruits[r], r);
            }            
            if (map.size() > 2) {
                // 如[1,0,1,4]，右指针指到第三个位置4的时候
                // 此时有三种种类，要去掉前面两个种类之一
                // 因为返回最长子串，根据最后一次出现的位置，所以去掉最后一次出现靠前的那一种类
                Set<Integer> keys = map.keySet();
                int idx = n + 1;
                for (Integer key : keys) {
                    if(key == fruits[r]){
                        continue;
                    }
                    idx = Math.min(idx, map.get(key));
                }
                l = idx + 1;
                map.remove(fruits[idx]);
            }
            ans = Math.max(ans, r - l + 1);
        }
        return ans;
    }
}