LeetCode904. 水果成篮 Java 滑动窗口法

一陸向北

已于 2023-11-04 19:23:27 修改

阅读量84

点赞数

分类专栏： LeetCode思路分享 # 数组文章标签：算法数据结构 java leetcode

于 2023-10-28 21:34:47 首次发布

本文链接：https://blog.csdn.net/m0_61195936/article/details/134096654

版权

LeetCode思路分享同时被 2 个专栏收录

14 篇文章 0 订阅

订阅专栏

数组

6 篇文章 0 订阅

订阅专栏

题目描述

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

AC代码以及详细注释

class Solution {
    public  int totalFruit(int[] fruits) {
        int left = 0;   // 左边界
        int right = 0;  // 右边界
        int leftFruit = -1; // 左手水果的种类
        int rightFruit = -1;// 右手水果的种类
        int nextLeft = 0;   // 下一次更新时左边界的位置
        int ret = 0;    // 最终结果
        int curSum = 0; // 每次换手中水果种类时结果处理的中间结果，如果比ret大就将ret更新为curSum

        for (right = 0; right < fruits.length; right++) {


            // 篮子的初始化，先把两个篮子放上不同的水果
            if (leftFruit == -1) {
                leftFruit = fruits[right];
            } else if (rightFruit == -1 && leftFruit != fruits[right]) {   // 当右篮为空且左篮的水果不等于当前右边界的水果时更新右篮水果
                rightFruit = fruits[right];
            }

            // 右边界遍历时遇到了不同于两个篮子中水果的种类
            if ((fruits[right] != leftFruit && fruits[right] != rightFruit) && leftFruit != -1 && rightFruit != -1) {
                // 将left向右遍历nextLeft
                while (left != nextLeft) {
                    left++;
                    curSum--;
                }
                // 更新左篮子或者右篮子中的水果种类
                if (leftFruit == fruits[left]) {
                    rightFruit = fruits[right];
                } else {
                    leftFruit = fruits[right];
                }
            }

            // 每向右遍历一次，必定要加一个curSum，因为每个水果都会被考虑。如果这个水果属于左右篮子水果的种类，则curSum直接俄加1，
            // 若不属于，则处理后减掉要丢弃的水果的数量后curSum加1
            curSum++;

            // 进行中间结果处理
            if (curSum > ret) {
                ret = curSum;
            }

            // 更新nextLeft的值, 如果当前右边界的水果与上一个水果种类不同，说明下一次更新篮中水果种类时要丢弃上一个种类的水果
            if (right == 0 || fruits[right] != fruits[right - 1]) {
                nextLeft = right;
            }
        }

        return ret;     // 最终返回ret作为最终结果
    }
}