原题链接
这道题就是典型的拓扑排序问题。简单说一下拓扑排序:对一个有向无环图G进行拓扑排序,是将G中所有顶点排成一个线性序列,使得图中任意一对顶点u和v,若边(u,v)∈E(G),则u在线性序列中出现在v之前。
拓扑排序算法主要是循环执行以下两步,直到不存在入度为0的顶点为止。
(1) 选择一个入度为0的顶点并输出之;
(2) 从网中删除此顶点及所有出边。
循环结束后,若输出的顶点数小于网中的顶点数,则说明存在回路,
回到这道题中,首先需要遍历一次pre中所有的有向边,统计每一个点的入度,为此我使用了map;并且记录每一个起始点的邻接点。
先将入度为0的节点进栈,然后进行广度优先遍历,将栈顶的元素保存到栈res中,并依次将栈顶的邻接点的入度减1,如果邻接点的入度为0,那就将邻接点压入栈中,直到栈空。
最后判断res的长度是否等于n,如果相等,那就说明有向图中不存在回路,否则存在。
代码如下:
class Solution {
public:
vector<int> findOrder(int n, vector<vector<int>>& pre) {
vector<int> res,tmpres;
vector<int> degree;
vector<set<int>> neighbor(n,set<int>());
for(auto item:pre){
degree[item[0]]++;
neighbor[item[1]].insert(item[0]);
}
for(int i=0;i<n;i++){
if(degree[i]==0){
tmpres.push_back(i);
}
}
while(tmpres.size()){
int t = tmpres.back();tmpres.pop_back();
res.push_back(t);
for(auto child:neighbor[t]){
degree[child]--;
if(degree[child]==0)tmpres.push_back(child);
}
}
return res.size()==n?res:vector<int>();
}
};