Leetcode 1557.可以到达所有点的最少点数目（Minimum Number of Vertices to Reach All Nodes）

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

1 题目描述（leetcode题目链接）

  给你一个 有向无环图 ， n 个节点编号为 0 到 n-1 ，以及一个边数组 edges ，其中 $edges[i]=[fro{m}_i,t{o}_i]$ 表示一条从点 $fro{m}_i$ 到点 $t{o}_i$ 的有向边。

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

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

输入：n = 6, edges = [[0,1],[0,2],[2,5],[3,4],[4,2]]
输出：[0,3]
解释：从单个节点出发无法到达所有节点。从 0 出发我们可以到达 [0,1,2,5] 。从 3 出发我们可以到达 [3,4,2,5] 。所以我们输出 [0,3] 。

2 题解

  从答案肯定是所有出发点（出度不为0的点）集合的子集，去除掉可从其他出发点到达的点即可。

class Solution:
    def findSmallestSetOfVertices(self, n: int, edges: List[List[int]]) -> List[int]:
        beg = set()
        end = set()
        for u, v in edges:
            beg.add(u)
            end.add(v)
        return list(beg - end)