【LeetCode】水果成篮 Java 题解

作者：Albert

2022-10-19
北京
本文字数：1184 字
阅读完需：约 1 分钟

题目描述

你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示，其中 fruits[i] 是第 i 棵树上的水果种类。

你想要尽可能多地收集水果。然而，农场的主人设定了一些严格的规矩，你必须按照要求采摘水果：

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

示例 1：
输入：fruits = [1,2,1]输出：3解释：可以采摘全部 3 棵树。示例 2：
输入：fruits = [0,1,2,2]输出：3解释：可以采摘 [1,2,2] 这三棵树。如果从第一棵树开始采摘，则只能采摘 [0,1] 这两棵树。示例 3：
输入：fruits = [1,2,3,2,2]输出：4解释：可以采摘 [2,3,2,2] 这四棵树。如果从第一棵树开始采摘，则只能采摘 [1,2] 这两棵树。示例 4：
输入：fruits = [3,3,3,1,2,1,1,2,3,3,4]输出：5解释：可以采摘 [1,2,1,1,2] 这五棵树。

复制代码

思路分析

今天的算法题目是数组题目。今天的题目描述比较长，需要一定的阅读理解能力。我们分析题目，题目描述之后两个篮子,每个篮子只能装一种水果，数量没有上限。每棵树也之后一种水果，求两种水果的最大的数目。
理解清楚题目之后，我们可以将问题转化成为滑动窗口的最长长度。这个滑动的窗口只能有 2 中类型的元素，我们可以使用 Map 这种数据结构来记录水果的类型和数量。从数组下标 0 开始，进行处理。将水果种类和数量记录到下标中，当 map 中元素的种类大于 2 的时候，我们可以逐步移动滑动窗口，保持滑动窗口的元素种类最多是 2 种。然后动态更新长度的最大值即可。
具体实现代码如下，供参考。

通过代码

class Solution {    public int totalFruit(int[] fruits) {        int n = fruits.length;        Map<Integer, Integer> cnt = new HashMap<Integer, Integer>();
        int left = 0, ans = 0;        for (int right = 0; right < n; ++right) {            cnt.put(fruits[right], cnt.getOrDefault(fruits[right], 0) + 1);            while (cnt.size() > 2) {                cnt.put(fruits[left], cnt.get(fruits[left]) - 1);                if (cnt.get(fruits[left]) == 0) {                    cnt.remove(fruits[left]);                }                ++left;            }            ans = Math.max(ans, right - left + 1);        }        return ans;    }}