携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第25天，点击查看活动详情

前言

小白算法比较菜，希望能激励我每日更新，从leetcode第一题开始，2022年目标300题，记录从0到1的全过程！！

904. 水果成篮

这是102场周赛的第二题

904. 水果成篮

现在介绍这个题。

题目

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

力扣有些题的题目真的是描述的有点难懂，不过确实，这也是程序员需要掌握的一种能力，讲以后工作中遇到的业务抽丝剥茧简化。

这个题就可以理解为求只包含两种元素的最长连续子序列。

2.解法一

解析

双指针，用一个map来储存已经收了的水果，以及对应的水果数量，时间复杂度算是O(2n)吧，但是可能是map的操作过多，导致耗时有点高。

class Solution {
    public int totalFruit(int[] fruits) {
        HashMap<Integer,Integer> map =new HashMap<>(); //key-水果 value-水果数量
        int l = 0;
        int r = 0;
        int max = 0;
        while(r < fruits.length) {
            if(map.containsKey(fruits[r])) {
                while(r < fruits.length && map.containsKey(fruits[r])) {
                    map.put(fruits[r],map.get(fruits[r])+1);
                    r++;
                }
            } else { //如果没有拿这个水果
                if(map.size()<2){  //如果只拿了0或者1种水果
                    map.put(fruits[r],1);
                    r++;
                } else {  //如果拿了两种水果
                    while(l<r) {
                        map.put(fruits[l],map.get(fruits[l])-1);
                        if(map.get(fruits[l])==0) {
                            map.remove(fruits[l]);
                            l++;
                            break;
                        }
                        l++;
                    }
                    
                } 
            }
            max = Math.max(max,r - l);
        }
        return max;
    }
}

提交排名

2.解法二

也是双指针，这里用两个变量leftFurits和rightFurits记录已经拿了的水果，那么最关键的问题是，当right指向一个新的水果时，该怎么修改left的值，以及修改leftFurits和rightFurits的值。

当right指向一个新水果时，right-1就是需要留下的那个老水果，从right-1往前遍历，找到不等于这个老水果的值就可以了，此时的指针就算left。

class Solution {
    public int totalFruit(int[] fruits) {
        int res = 0;
        int left = 0;
        int right = 0;
        int leftFurits = fruits[left];
        int rightFurits = fruits[right];
        while(right < fruits.length) {
            while(right < fruits.length && (fruits[right] == leftFurits || fruits[right] == rightFurits)) {
                right++;
            }
            if(right == fruits.length) return Math.max(res,right - left);
            res = Math.max(res,right - left);
            left = right - 1;
            leftFurits = fruits[left];
            if(right!=fruits.length)  rightFurits = fruits[right];
            while(left >= 0 && fruits[left]==leftFurits) {
                left--;
            }
            left++;
            res = Math.max(res,right - left);
        }
        return res;
    }
}

提交排名

3.结束

这个题把题目看懂了还算是很好理解的，周赛第二题的话，除非是某次的实在是太难，一般我的水平就是第二题能做出来，想先做一个偶尔三题的选手，再慢慢变成每次都三题的选手，还是要加强训练啊，gogogo，刷题刷题，每天一道，三年1000道！！！！