题目
给你一个 有向无环图 , 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] += 1for 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))