0015【数组】中等LeetCode904.水果成篮

妙语空空

已于 2023-02-02 10:39:57 修改

阅读量61

点赞数

分类专栏： 152.0LeetCode 文章标签：算法 leetcode java 数据结构

于 2023-02-02 09:45:20 首次发布

本文链接：https://blog.csdn.net/MUYUHF/article/details/128844698

版权

152.0LeetCode 专栏收录该内容

30 篇文章 0 订阅

订阅专栏

你想要尽可能多地收集水果。然而，农场的主人设定了一些严格的规矩，你必须按照要求采摘水果：
①你只有两个篮子，并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
②你可以选择任意一棵树开始采摘，你必须从每棵树（包括开始采摘的树）上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次，你将会向右移动到下一棵树，并继续采摘。
③一旦你走到某棵树前，但水果不符合篮子的水果类型，那么就必须停止采摘。

给你一个整数数组 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 <= 10^5
0 <= fruits[i] < fruits.length

2.题目解答

方法一：最大滑动窗口

解题思路
1.最大滑窗：给定数组nums，定义滑窗的左右边界i、j，求满足某个条件的滑窗的最大长度。
while (j < len(nums)):{
    判断[i, j]是否满足条件
    while 不满足条件：{
        i += 1 （最保守的压缩i，一旦满足条件了就退出压缩i的过程，使得滑窗尽可能的大）}
    不断更新结果（注意在while外更新！）
    j += 1
}

public class Solution {
    public int totalFruit(int[] fruits) {
        int result = 0;
        int type1 = -1;
        int type2 = -1;
        int start = 0;
        for (int end = 0; end < fruits.length; end++) {
            // 初始化水果类型1
            if (type1 == -1) {
                type1 = fruits[end];
                // 初始化水果类型2
            } else if (fruits[end] != type1 && type2 == -1) {
                type2 = fruits[end];
                // 如果遇到第3种水果类型，就更新前2种水果类型：
                // 水果类型1赋值为当前位置前一棵树的水果类型
                // 水果类型2赋值为当前水果类型，即第3种水果类型
            } else if (fruits[end] != type1 && fruits[end] != type2) {
                type1 = fruits[end - 1];
                type2 = fruits[end];
                for (int i = end - 2; i >= start; i--) {
                    if (fruits[i] != type1) {
                        start = i + 1;
                        break;
                    }

                }
            }
            result = Math.max(result, end - start + 1);
        }
        return result;
    }
}