持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第20天，点击查看活动详情

刷题的日常-2022年10月17号

一天一题，保持脑子清爽

水果成篮

来自leetcode的 904 题，题意如下：

你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示，其中 fruits[i] 是第 i 棵树上的水果 种类 。
你想要尽可能多地收集水果。然而，农场的主人设定了一些严格的规矩，你必须按照要求采摘水果：

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

给你一个整数数组 fruits ，返回你可以收集的水果的 最大 数目。

理解题意

通过题意，我们可以将信息整理如下：

• 题目给定一个数组
• 要求我们以连续的两个不同值为一组进行统计
• 返回只有两种水果的连续的最大值

做题思路

因为只有两个值，所以我们可以直接用双指针的方法进行操作，步骤如下：

• 先找出第一种水果的值
• 如果到达数组最后，返回结果
• 如果没有，记录第二种水果的开始位置
• 继续往后遍历数组
• 如果和第二个值相同，记录最大值
• 如果和第一个值相同，更新第二个索引为当前索引，第一个索引更新为第二个索引，重置统计值
• 返回当前统计值和之前统计结果的最大值

代码实现

代码实现如下，因为只需要扫描一次数组，所以事件复杂度为O(n):

public class Solution {
    public int totalFruit(int... fruits) {
        int result, f = 0, s, idx = 1, curr = 1;
        while (idx < fruits.length && fruits[idx] == fruits[0]) {
            idx++;
            curr++;
        }
        result = curr;
        if (idx == fruits.length) {
            return result;
        }
        s = idx;
        while (idx < fruits.length) {
            if (fruits[idx] == fruits[s]) {
                idx++;
                curr++;
                continue;
            }
            if (fruits[idx] == fruits[f]) {
                curr++;
                f = s;
                s = idx++;
                continue;
            }
            result = Math.max(curr, result);
            curr = idx - s + 1;
            f = s;
            s = idx++;
        }
        return Math.max(curr, result);
    }
}