关于后端:904-水果成篮-简单滑动窗口运用题

3次阅读

共计 1833 个字符，预计需要花费 5 分钟才能阅读完成。

这是 LeetCode 上的 904. 水果成篮 ，难度为中等。

Tag :「双指针」、「模仿」、「滑动窗口」

你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 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 <= 10^5$
$0 <= fruits[i] < fruits.length$

这次是真看样例，都了解不了是啥题意：其实就是从任意地位开始，同时应用 两个篮子采集，一旦抉择后不能批改篮子所装的水果品种，当所有树解决完或遇到第一棵品种不同的树则进行。

滑动窗口模拟题：应用 j 和 i 别离代表滑动窗口的两端，窗口品种不超过 $2$ 种为非法。

Java 代码：

class Solution {public int totalFruit(int[] fs) {
        int n = fs.length, ans = 0;
        int[] cnts = new int[n + 10];
        for (int i = 0, j = 0, tot = 0; i < n; i++) {if (++cnts[fs[i]] == 1) tot++;
            while (tot > 2) {if (--cnts[fs[j++]] == 0) tot--;
            }
            ans = Math.max(ans, i - j + 1);
        }
        return ans;
    }
}

TypeScript 代码：

function totalFruit(fs: number[]): number {
    let n = fs.length, ans = 0
    const cnts = new Array<number>(n + 10).fill(0)
    for (let i = 0, j = 0, tot = 0; i < n; i++) {if (++cnts[fs[i]] == 1) tot++
        while (tot > 2) {if (--cnts[fs[j++]] == 0) tot--
        }
        ans = Math.max(ans, i - j + 1)
    }
    return ans
}

Python 代码：

class Solution:
    def totalFruit(self, fs: List[int]) -> int:
        n, ans = len(fs), 0
        j, tot = 0, 0
        cnts = defaultdict(int)
        for i in range(n):
            cnts[fs[i]] += 1
            if cnts[fs[i]] == 1:
                tot += 1
            while tot > 2:
                cnts[fs[j]] -= 1
                if cnts[fs[j]] == 0:
                    tot -= 1
                j += 1
            ans = max(ans, i - j + 1)
        return ans