Leetcode 1298:你能从盒子里获得的最大糖果数(超详细的解法!!!)
in leetcode with 0 comment

Leetcode 1298:你能从盒子里获得的最大糖果数(超详细的解法!!!)

in leetcode with 0 comment

给你 n 个盒子,每个盒子的格式为 [status, candies, keys, containedBoxes] ,其中:

给你一个 initialBoxes 数组,表示你现在得到的盒子,你可以获得里面的糖果,也可以用盒子里的钥匙打开新的盒子,还可以继续探索从这个盒子里找到的其他盒子。

请你按照上述规则,返回可以获得糖果的 最大数目

示例 1:

输入:status = [1,0,1,0], candies = [7,5,4,100], keys = [[],[],[1],[]], containedBoxes = [[1,2],[3],[],[]], initialBoxes = [0]
输出:16
解释:
一开始你有盒子 0 。你将获得它里面的 7 个糖果和盒子 1 和 2。
盒子 1 目前状态是关闭的,而且你还没有对应它的钥匙。所以你将会打开盒子 2 ,并得到里面的 4 个糖果和盒子 1 的钥匙。
在盒子 1 中,你会获得 5 个糖果和盒子 3 ,但是你没法获得盒子 3 的钥匙所以盒子 3 会保持关闭状态。
你总共可以获得的糖果数目 = 7 + 4 + 5 = 16 个。

示例 2:

输入:status = [1,0,0,0,0,0], candies = [1,1,1,1,1,1], keys = [[1,2,3,4,5],[],[],[],[],[]], containedBoxes = [[1,2,3,4,5],[],[],[],[],[]], initialBoxes = [0]
输出:6
解释:
你一开始拥有盒子 0 。打开它你可以找到盒子 1,2,3,4,5 和它们对应的钥匙。
打开这些盒子,你将获得所有盒子的糖果,所以总糖果数为 6 个。

示例 3:

输入:status = [1,1,1], candies = [100,1,100], keys = [[],[0,2],[]], containedBoxes = [[],[],[]], initialBoxes = [1]
输出:1

示例 4:

输入:status = [1], candies = [100], keys = [[]], containedBoxes = [[]], initialBoxes = []
输出:0

示例 5:

输入:status = [1,1,1], candies = [2,3,2], keys = [[],[],[]], containedBoxes = [[],[],[]], initialBoxes = [2,1,0]
输出:7

提示:

解题思路

这个问题实际上非常简单,描述的过程比较复杂。我们读懂题意,不难发现实际上所有的盒子是一种递归嵌套的关系,实际上就是一颗树,所以很容易想到通过dfsbfs遍历,但是这个问题有一个特殊的地方(通过钥匙开盒子),所以对于能不能打开盒子的问题,需要两个条件:(1)有钥匙 (2)可以遍历到该盒子。

为了处理问题更加简洁,我们约定当拥有钥匙的时候,就将对应盒子打开,那么现在就只需要一个条件了(盒子能否遍历到)。

我这里使用bfs去遍历(这样可以不用开辟一个visited数组,用于记录那些元素访问过)。对于不能打开的盒子,我们不应该立即去除(因为后面的盒子中可能有该盒子的钥匙),而是将其放到队尾。

那么遍历结束的条件是什么呢?如果队列中的盒子都无法打开的话,就结束遍历。具体实现的时候我们当然不用遍历队列中的每个盒子,并判断其是否可以打开。我们只需要通过一个变量记录上一层遍历过程中是否有盒子打开即可。

class Solution:
    def maxCandies(self, status: List[int], candies: List[int], keys: List[List[int]], containedBoxes: List[List[int]], initialBoxes: List[int]) -> int:
        q, res = initialBoxes, 0
        while q:
            q_len, changed = len(q), False
            for _ in range(q_len):
                b = q.pop(0)
                if status[b]:
                    changed = True
                    res += candies[b]
                    q += containedBoxes[b]
                    for i in keys[b]:
                        status[i] = 1
                else:
                    q.append(b)

            if not changed:
                return res
        return res

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题,希望大家指出!!!

「如果我的文章对你有很大帮助,那么不妨~!」

coordinate

谢谢老板O(∩_∩)O~

使用微信扫描二维码完成支付

Responses