题解 | #牛牛吃水果的顺序#

牛牛吃水果的顺序

https://www.nowcoder.com/practice/336b43ff1d664cdd8e3e39acb67dfa39

#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
#
# @param numFruits int整型
# @param prerequisites int整型二维数组
# @return int整型二维数组
#
class Solution:
    def findFruitOrder(
        self, numFruits: int, prerequisites: List[List[int]]
    ) -> List[List[int]]:
        # write code here
        if numFruits == 0:
            return []
        if numFruits == 1:
            return [[0]]
        pre_dict = dict()
        for pre in prerequisites:
            if pre[0] in pre_dict:
                pre_dict[pre[0]].add(pre[1])
            else:
                pre_dict[pre[0]] = set()
                pre_dict[pre[0]].add(pre[1])

        ans = []

        def dfs(pre_order: list, used: set, unused: list, requirement: dict):
            if len(pre_order) == numFruits:
                ans.append(pre_order)
                return
            for i, f in enumerate(unused):
                if not f in requirement or requirement[f] <= used:
                    dfs(
                        pre_order + [f],
                        used | set([f]),
                        unused[:i] + unused[i + 1 :],
                        requirement,
                    )
            return

        dfs([], set(), list(range(numFruits)), pre_dict)
        ans.sort()
        return ans

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-09 16:15
我应届生,去年10月份开始在这家公司实习,到今年10月份正好一年想(实习+试用期),在想要不要提前9月份就离职,这样好找工作些,但又差一个月满一年,又怕10月份国庆回来离职,容易错过了下半年的金九银十,到年底容易gap到年后
小破站_程序员YT:说这家公司不好吧,你干了快一年 说这家公司好吧,你刚毕业就想跑路说你不懂行情吧,你怕错过金九银十说 你懂行情吧,校招阶段在实习,毕业社招想换工作 哥们,我该怎么劝你留下来呢
应届生,你找到工作了吗
点赞 评论 收藏
分享
06-17 21:57
门头沟学院 Java
白友:噗嗤,我发现有些人事就爱发这些,明明已读不回就行了,就是要恶心人
点赞 评论 收藏
分享
06-04 09:27
门头沟学院 Java
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务