python爬虫笔记1

使用桶排序算法中的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 []