Leetcode1557 Minimum Number of Vertices to Reach All Nodes [Python]

用了一点拓扑排序的思想,就是得先确定每个节点的出度和入度。始发节点一定是入度为0的节点,随后需要对始发节点的出度做排序,出度越高越可能把全部节点遍历完,所以优先开始bfs。

class Solution:
    def findSmallestSetOfVertices(self, n: int, edges: List[List[int]]) -> List[int]:
        indegree = [0 for _ in range(n)]
        outdegree = [0 for _ in range(n)]
        graph = collections.defaultdict(list)
        for a, b in edges:
            graph[a].append(b)
            outdegree[a] += 1
            indegree[b] += 1
        meet = set()
        
        def bfs(node, meet):
            que = collections.deque()
            que.append(node)
            while que:
                size = len(que)
                for _ in range(size):
                    curnode = que.popleft()
                    for nextnode in graph[curnode]:
                        if nextnode not in meet:
                            meet.add(nextnode)
                            que.append(nextnode)
        que = []                    
        for node in range(len(indegree)):
            if indegree[node] == 0:
                heapq.heappush(que,(-outdegree[node], node))
        res = [] 
        while que and len(meet) != n:
            _, node = heapq.heappop(que)
            bfs(node, meet)
            res.append(node)
        return res      
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值