1557. 可以到达所有点的最少点数目

Posted on | In | |

给你一个 有向无环图 ， n 个节点编号为 0 到 n-1 ，以及一个边数组 edges ，其中 edges[i] = [fromi, toi] 表示一条从点 fromi 到点 toi 的有向边。

找到最小的点集使得从这些点出发能到达图中所有点。题目保证解存在且唯一。

你可以以任意顺序返回这些节点编号。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/minimum-number-of-vertices-to-reach-all-nodes
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

最开始的想法中，vertext并未转换为set类型，判断edge[1]是否在vertex中，如果在则删除，但是超时了。后来换成把入度非0的节点和vertex都换成集合就过了。。

1
2
3
4
5
6
7
8
9
10
11
12
class Solution(object):
    def findSmallestSetOfVertices(self, n, edges):
        """
        :type n: int
        :type edges: List[List[int]]
        :rtype: List[int]
        """
        vertex = set(list(range(n)))
        indegree = set()
        for edge in edges:
            indegree.add(edge[1])
        return list(vertex - indegree)