力扣904.水果成篮

最新推荐文章于 2024-09-30 14:51:11 发布

交通组织之虎

最新推荐文章于 2024-09-30 14:51:11 发布

阅读量133

点赞数

文章标签： leetcode 算法

本文链接：https://blog.csdn.net/s4444sss/article/details/125189530

版权

904.水果成篮

思路：

本题使用滑动窗口方法。当有三个不同的时候就将此窗口最左边的移出，然后依次找最多的即可。

代码实现：

class Solution {
    public int totalFruit(int[] fruits) {
        HashMap<Integer, Integer> map = new HashMap<>();
        int left = 0;
        int right = 0;
        int ans = 0;
        int length = fruits.length;

        while (right < length) { //滑动窗口
            map.put(fruits[right], map.getOrDefault(fruits[right], 0) + 1);
            while (map.size() > 2) { //有三个不相同的
                map.put(fruits[left], map.getOrDefault(fruits[left], 0) - 1);//移除左边的
                //此处不能直接remove，因为考虑到  [3,3,3,1,...]这种情况
                if (map.get(fruits[left]) == 0) {
                    map.remove(fruits[left]);
                }
                left++;
            }
            //求ans放在外面，因为如果放在while里面的话，[1,2,1]这种情况不会求ans
            ans = Math.max(right - left + 1, ans);
            right++;
        }
        return ans;
    }
}