如果直接使用邻接表不记录已经遍历过的数,暴力搜索超超时,所以需要拓扑排序来记录入度为0的点,入度为0时,他之前的父节点已经记录完毕,可以直接加入到他的子节点中。
最后还需要转换结果,从小到大,直接用set可以解决,注意,一个点可能经过不同的路径到达另外一个点,这样父节点可能插入两个相同的数值,set正好解决了这个问题,同时也可以排序,就是用的空间可能比unorder_set大。
class Solution {
public:
vector<vector<int>> getAncestors(int n, vector<vector<int>>& edges) {
vector<int>h[n]; //邻接表
vector<int>index(n); // 入读表 注意一个是二维数组一个是一维数组
vector<set<int>> ans(n); // 每个点之前有几个, 用set主要是可以直接删除有的相同元素
for (auto e : edges) {
index[e[1]]++;
h[e[0]].push_back(e[1]);
}
queue<int> q; // 宽搜拓扑排序找出更新结束的点
for (int i = 0; i < n; i++)
if (!index[i]) q.push(i);
while (q.size()) {
int u = q.front();
q.pop();
for (int i = 0; i < h[u].size(); i++) {
int x = h[u][i];
ans[x].insert(u);
for (auto k : ans[u]) {
ans[x].insert(k);
}
index[x]--;
if (!index[x]) {
q.push(x);
}
}
}
vector<vector<int>> res(n);
for (int i = 0; i < n; i++) {
for (auto a : ans[i]) {
res[i].push_back(a);
}
}
return res;
}
};