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

水果成篮

题目

904. 水果成篮

难度中等337收藏分享切换为英文接收动态反馈

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

分析

简单分析题意，这次的题目真的是看半天，费劲！！！简单来说就是找到一个最长的子数组，这个子数组只有两种字。返回最长的长度即可

暴力解法

先贴上代码

/**
 * @param {number[]} fruits
 * @return {number}
 */
var totalFruit = function (fruits) {
    if (fruits.length === 1) {
        return 1;
    }
    let i = 0;
    let type1;
    let type2;
    let res = 0;
    while (i < fruits.length - 1) {
        let j = i + 1;
        let n = 1;
        type1 = fruits[i];
        while (fruits[j] === type1) {
            n++;
            j++;
        }
        if (j === fruits.length) return res === 0 ? j : res;
        type2 = fruits[j];
        while (j < fruits.length) {
            if (fruits[j] === type1 || fruits[j] === type2) {
                n++;
                j++;
            } else {
                break;
            }
        }
        if (n > res) {
            res = n;
        }
        // j要么时位于length处要么是位于第三个种类处
        if (j < 2) {
            i++;
        } else {
            // /否则就是j的值>=2,那么有三个以上的数，i为1\0，i应该变为1或者0 
            if (fruits[j - 1] !== fruits[j - 2]) {
                i = j - 1;
            }
            else {
                while (fruits[j - 1] === fruits[j - 2]) {
                    i = j - 2;
                    j--;
                }
            }
        }
    }
    return res;
};

其实最暴力的解法就是啥都不管，就遍历fruit的长度，以每个下标都找一个最长的长度，取即可，但是你提交就会发现给你一个很长的测试案例，结果超时！。

因此我们需要优化，我上面这个代码就是优化过的

思路很简单，就是不必全部遍历，只要从上次结束的最后的重复元素的开头处开始就可以了。也就是说我们取上次结果 的 最后一个重复的字符串 ，从这个字符串的首字符开始，因为再往前取也不可能超过上次的结果。

我的代码就在最后进行while循环往后找下一个起始位置，能够找到然后再循环。

滑动窗口

当然 我上面的代码是又臭又长！！，其实官方题目已经在多处暗示你用滑动窗口解决了。我一开始也想到了这个办法，但是我想着什么题写个暴力解挺不错的，结果就写出了个山。。。

话不多说，刷个几个题的应该都知道滑动窗口，就是left与right组成窗口，移动窗口来寻找最大的窗口。

var totalFruit = function (fruits) {
    // 题目的各种提示都在让你往滑动窗口上靠
    const n = fruits.length;
    const cnt = new Map();
    // cnt 键是种类，值是数量，
    let left = 0, ans = 0;
    for (let right = 0; right < n; ++right) {
        //右边界一直走
        cnt.set(fruits[right], (cnt.get(fruits[right]) || 0) + 1);
        // 设置键的值加一
        while (cnt.size > 2) {
            // 当map里面有超过两个种类时 我们需要把从left的元素开始处删除，删除到left到right的窗口只有两种。
            cnt.set(fruits[left], cnt.get(fruits[left]) - 1);
            // 键的值减一
            if (cnt.get(fruits[left]) == 0) {
                cnt.delete(fruits[left]);
            }  //然后删除left的键值对
            ++left;
        }
        ans = Math.max(ans, right - left + 1);
        //取到长度即可
    }
    return ans;
};

很明显，滑动窗口的代码简洁很多，另外就是map能更方便。

其实优化思路都是一致的，找到最后重复字串的起始位置就可以优化时间了。

总结

本次文章的难点在于善用滑动窗口解决寻找最长的问题。

结语

本次的文章到这里就结束啦！♥♥♥读者大大们认为写的不错的话点个赞再走哦 ♥♥♥

每天一个知识点，每天都在进步！♥♥