- 拓扑排序原理:对有向无环图DAG的顶点进行排序,使得对于每一条有向边(u,v),均有u(在排序记录中)比v先出现,亦可理解为对某点v而言,只有当v的所有源点均出现了v才能出现
根据定义,可以得到以下两个结论:
- 如果图G中存在环(即G不是有向无环图),那么G不存在拓扑排序,反之如果G存在拓扑排序,则G中没有环
- 如果G是有向无环图,那么他的拓扑排序可能不止一种
我们用两种方法来求解拓扑排序:dfs和bfs
方法一 :深度优先搜索
- 我们用一个栈来存储所有已经搜索完成的结点,对于某个结点,他在搜索过程中有三种状态:
[未搜索]:还没有搜索到这个结点
[搜索中]:已经搜索过这个结点了,但是还没有回溯到该节点,即该节点还没有入栈,还有相邻结点没有搜索完成
[已完成]:已经搜索过了这个结点并且已经回溯过,即该结点已经入栈,并且所有该节点的相邻节点都出现在栈的更底部的位置,满足拓扑排序的要求。
- 在每一轮的搜索搜索开始时,我们任取一个「未搜索」的节点开始进行深度优先搜索。当搜索到当前结点u时,我们将其置为[搜索中],遍历其相邻结点v:
- 如果v为[未搜索],我们开始搜索v,直到搜索回溯到u
- 如果v为[搜索中],说明存在环,不存在拓扑排序
- 如果v为[已完成],说明v已经在栈中了,对结果不影响,不用进行任何操作。
- 当所有的结点都被搜索完成后,如果没有找到环,那么栈中存储的结点,从栈顶到栈底的顺序即为拓扑排序
vector<vector<int>>edges; //邻接表
vector<int>vis; //存储状态
vector<int>ans; //记录答案
bool res=true; //判断是否有环
void dfs(int u){
vis[u]=1; //标记为搜索中
for(auto& v:edges[u]){ //搜索所有邻接的点
if(vis[v]==1){ //有环
res=false;
return ;
}
else if(vis[v]==0){ //没遍历过,搜索
dfs(v);
if(res==false) return ;
}
}
vis[u]=2; //标记为搜索完成
ans.push_back(u); //入栈
}
vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
vis.resize(numCourses,0);
edges.resize(numCourses);
for(auto& num:prerequisites){
edges[num[1]].push_back(num[0]); //创建邻接表
}
for(int i=0;i<numCourses;i++){
if(!vis[i]) dfs(i); //从没有被遍历过的结点开始
}
if(!res) return {};
reverse(ans.begin(),ans.end()); //模拟栈
return ans;
}
方法二:广度优先搜索+入度判断
- 我们使用一个队列来进行广度优先搜索。开始时,所有入度为 0 的节点都被放入队列中,它们就是可以作为拓扑排序最前面的节点,并且它们之间的相对顺序是无关紧要的。
- 在广度优先搜索的每一步中,我们取出队首的节点 u:
- 将u放入答案中
- 我们移除u的所有出边,也就是将u的所有相邻节点的入度减少1。如果某个相邻节点v的入度变为0,那么我们就将v放入队列中
- 在广度优先搜索的过程结束后。如果答案中包含了这n个节点,那么我们就找到了一种拓扑排序,否则说明图中存在环,也就不存在拓扑排序了。
vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
vector<vector<int>>edges(numCourses);
vector<int>degree(numCourses,0); //存储结点的入度
queue<int>q;
vector<int>ans;
for(auto& num:prerequisites){
edges[num[1]].push_back(num[0]);
degree[num[0]]++; //入度++
}
for(int i=0;i<numCourses;i++){
if(degree[i]==0) q.push(i); //将入度为0的放入队列中
}
while(!q.empty()){
int cur=q.front();
q.pop();
ans.push_back(cur);
for(auto& v:edges[cur]){
degree[v]--; //相邻的结点入度减一
if(degree[v]==0) q.push(v); //如果入度为0,放入队列中
}
}
if(ans.size()!=numCourses) return {};
return ans;
}