LeetCode 课程表二(拓扑排序+Python)

使用桶排序算法中的kahn(卡恩)算法,也可以使用dfs。

这里使用卡恩算法,主要维护一个列表cnt,cnt【i】表示能到达节点i的边,比如说:a到c有一条边,b到c有一条边,那么cnt【c】= 2。当然,如果从a走到c一次,cnt【i】= 1。如果b再到cc一次,那么cnt【i】= 0。如果某个节点i,cnt【i】= 0,说明没有边能到达i节点,就是说i节点前面没有其他节点了,那么可以选择i节点。

class Solution:
    def findOrder(self, numCourses: int, prerequisites: List[List[int]]) -> List[int]:
        ans = []
        # kahn 算法
        edge = defaultdict(list)
        cnt = [0]*numCourses
        for a, b in prerequisites:
            edge[b].append(a)
            cnt[a] += 1
        qu = deque()
        for i in range(numCourses):
            if cnt[i] == 0:
                qu.append(i)
        while qu:
            f = qu.popleft()
            ans.append(f)
            for nex in edge[f]:
                cnt[nex] -= 1
                if cnt[nex] == 0:
                    qu.append(nex)
        
        return ans if len(ans) == numCourses else []

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值