题目大意
给你一个 有向无环图 , 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的点,两边遍历即可。
class Solution {
public:
vector<int> findSmallestSetOfVertices(int n, vector<vector<int>>& edges) {
vector<int> isRoot(n, 0);
// 遍历一遍,看看哪些点的入度为0
for (auto & vec : edges){
isRoot[vec[1]]++;
}
vector<int> ans;
for (int i = 0; i < n; ++i){
if (isRoot[i] == 0){
ans.push_back(i);
}
}
return ans;
}
};