破解 leetcode 904：装水果的篮子 - 巧用滑动窗口法，篮满为盈

见解分享

2023-11-13 22:03:57

如何破解 LeetCode 904：装水果的篮子

破解 LeetCode 904

作为算法竞赛爱好者，我们经常面对各种难题，其中 LeetCode 904「装水果的篮子」就是一道令人头疼的题目。它要求我们找到最多不同种类的水果，同时满足特定限制条件。破解这道题的关键在于掌握强大的算法策略——滑动窗口法。

滑动窗口法

滑动窗口法是一种动态规划技术，用于解决寻找特定子数组的问题。它的本质是使用一个窗口在数组中移动，并在移动过程中不断更新窗口内的信息。通过这种方式，我们可以高效地找到满足特定条件的子数组。

破解步骤

解决 LeetCode 904 的具体步骤如下：

初始化窗口： 以窗口大小为 2，遍历数组，记录窗口内的不同水果种类数。
滑动窗口： 移动窗口，每移动一步，更新窗口内的信息：
- 如果窗口内水果种类数少于 2，则扩大窗口。
- 如果窗口内水果种类数达到 2，则缩小窗口，直到窗口内水果种类数减少为 1。
记录最大种类数： 在窗口移动过程中，记录窗口内不同水果种类数的最大值。

代码示例

为了加深理解，我们提供一个用 Java 编写的代码示例：

import java.util.HashMap;
import java.util.Map;

public class Solution {
    public int totalFruit(int[] fruits) {
        if (fruits == null || fruits.length == 0) {
            return 0;
        }

        int left = 0, right = 0, maxFruit = 0, count = 0;
        Map<Integer, Integer> fruitCount = new HashMap<>();

        while (right < fruits.length) {
            if (fruitCount.size() <= 2) {
                fruitCount.put(fruits[right], fruitCount.getOrDefault(fruits[right], 0) + 1);
                maxFruit = Math.max(maxFruit, ++count);
                right++;
            } else {
                while (fruitCount.size() > 2) {
                    int leftFruit = fruits[left];
                    fruitCount.put(leftFruit, fruitCount.get(leftFruit) - 1);
                    if (fruitCount.get(leftFruit) == 0) {
                        fruitCount.remove(leftFruit);
                    }
                    left++;
                    count--;
                }
            }
        }

        return maxFruit;
    }
}