题目描述
给你一个 有向无环图 , n
个节点编号为 0
到 n-1
,以及一个边数组 edges
,其中 edges[i] = [fromi, toi]
表示一条从点 fromi
到点 toi
的有向边。
找到最小的点集使得从这些点出发能到达图中所有点。题目保证解存在且唯一。
示例:
输入: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] 。
解题思路
题目明确指出,无环图、题目保证解存在且唯一,可以这么想。
- 入度为0的结点,必为图的入口;
- 入度不为0的结点,必可以由弧头结点访问
所以不难得出结论,题解就是入度为0的点的集合。因此,用一个一维数组记录各结点的入度,然后遍历数组,将入度为0的结点下标放入res[]
,遍历完返回即可。
代码实现
class Solution {
public:
vector<int> findSmallestSetOfVertices(int n, vector<vector<int>>& edges) {
vector<int> in(n);
for(auto& vec : edges){
in[vec[1]]++;
}
vector<int> res;
for(int i = 0; i < n; i++){
if(in[i] == 0) res.push_back(i);
}
return res;
}
};
运行结果: