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

题目

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

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

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

示例 1：

输入：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：

输入：n = 5, edges = [[0,1],[2,1],[3,1],[1,4],[2,4]]
输出：[0,2,3]
解释：注意到节点 0，3 和 2 无法从其他节点到达，所以我们必须将它们包含在结果点集中，这些点都能到达节点 1 和 4 。

思路

出入度统计

首先，观察示例可以发现答案最小集有两个规律：

• 答案必须包括入度为0的节点，如果没有包括，那么该节点肯定不能到达；

• 如果有一条a->b的边，那么b一定不会在答案当中，因为选a能到达b也能到达a

由于保证解存在且唯一，所以最小集即为图中所有入度为0的节点

python代码：

class Solution(object):
    def findSmallestSetOfVertices(self, n, edges):
        ans = []
        inDegree = [0 for _ in range(n)]
        for i,j in edges:
            inDegree[j] += 1
        for i in range(n):
            if inDegree[i] == 0:
                ans.append(i)
        return ans

由于python set 可以用来去重，可以统计edges 所有可到达的点，即第二个元素，然后从 0-n中剔除这部分即可：

class Solution(object):
    def findSmallestSetOfVertices(self, n, edges):
        return list(set(range(n)) - set(j for i,j in edges))