力扣904(Java)-水果成篮(中等)

发布时间 2023-04-25 10:58:03作者: 我不想一直当菜鸟

题目:

你正在探访一家农场,农场从左到右种植了一排果树。这些树用一个整数数组 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] 这五棵树。
 

提示:

  • 1 <= fruits.length <= 105
  • 0 <= fruits[i] < fruits.length

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/fruit-into-baskets
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路:

哈希表+滑动窗口

  • 用哈希表map来当做窗口来维护当前水果的种类以及数量;
  • 用双指针slow和fast来维护窗口的左右边界;
  • 遍历数组fruits,将当前水果加入哈希表中,且对应的数量也要加
    • 判断水果的种类大于2即map.size() > 2时:将左边界left对应种类的水果数量减1;如果左边界slow对应的水果数量等于0时,将左边界水果种类删除(key值删除);将左边界slow向右移动,直到种类小于2;
    • 更新最大长度
  • 返回最大长度;
 1 class Solution {
 2     public int totalFruit(int[] fruits) {
 3         Map<Integer, Integer> map = new HashMap<>();
 4         int slow = 0, sublen = 0, ans = 0;
 5         for (int fast = 0; fast < fruits.length; fast++){
 6             map.put(fruits[fast], map.getOrDefault(fruits[fast], 0) + 1);
 7             //如果水果种类大于2,就减少第一种水果的数量
 8             while (map.size() > 2){
 9                 map.put(fruits[slow], map.get(fruits[slow]) - 1);
10                 //如果水果数量为0,则移除当前种类
11                 if (map.get(fruits[slow]) == 0){
12                    map.remove(fruits[slow]);  
13                 }
14                 slow++;
15             }
16             //更新长度
17             sublen = fast - slow + 1;
18             ans = Math.max(sublen, ans);  
19         }
20         return ans;
21     }
22 }

最小滑动窗口模板:给定数组nums,定义滑动窗口左右边界slow 和 fast,求满足某个条件的滑窗的最小长度;

例如:力扣209.长度最小的子数组

1 while ( fast < nums.length){
2        while (满足条件){
3              不断更新结果,在While内更新;
4               slow++;
5      }
6        fast++;
7  }

最大滑动窗口模板:给定数组nums,定义滑动窗口左右边界slow 和 fast,求满足某个条件的滑窗的最大长度;

例如:力扣904.水果成篮

1 while ( fast < nums.length){
2       while (不满足条件){
3              slow++;
4      }
5       不断更新结果,在While外更新;
6       fast++;
7 }