拓扑排序
by 葛鑫
和之前谈到的拓扑排序不同,拓扑排序指的是将一个有向无环图展开成一条链。唯一的要求是,若存在边u→v,那么排序结果中u必定出现在v之前。并且拓扑排序大部分情况下会有很多种正确结果。
对这个有向无环图来说, {6,1,3,2,7,4,5,8} 和 {1,6,2,3,4,5,7,8} 这两种排序结果都是正确的。拓扑排序可以运用在很多优化问题当中,尤其是规划问题,很像那种小学奥数题,给你很多任务,每个任务耗费多少时间,是有没有前置要求,让你求完成所有任务耗费的最少时间,这里就不详细展开了,就说说如何进行拓扑排序。
最开始我看的算法导论,给出的伪代码如下。
TOPOLOGICAL-SORT.G/
.1 call DFS.G/ to compute finishing times :f for each vertex
.2 as each vertex is finished, insert it onto the front of a linked list
.3 return the linked list of vertices
对,没错,看完一头雾水,这什么东西,DFS(G)中的G是根节点吗?不知道根节点的话随便跑吗?对于多次访问到的节点,finishing time要怎么算呢?然后我看到了斯坦福的算法PPT,一下子舒坦了。
这个伪码第一眼看上去可能不好理解,但至少好实现的多。然后我就写出了源码,调试一下瞬间秒懂。
vector<vector<int>> v;
int n,m,a,b,deg[MAXN];
queue<int> Topological_Sort(){
queue<int> ans,q;
for(int i=1;i<=n;i++){
if(!deg[i])
q.push(i);
}
while(!q.empty()){
int u = q.front();
for(auto i=v[u].begin();i!=v[u].end();i++){
deg[*i]--;
if(!deg[*i])
q.push(*i);
}
v[u].clear();
ans.push(q.front());
q.pop();
}
return ans;
}
int main(){
cin>>n>>m;
memset(deg,0,sizeof(deg));
v.resize(n+1);
forn(i,m){
cin>>a>>b;
v[a].pb(b);
deg[b]++;
}
queue<int> ans = Topological_Sort();
while(!ans.empty()){
cout<<ans.front()<<" ";
ans.pop();
}
return 0;
}
首先我们可以肯定,如果一个结点的入度是0,那么它必然是根节点,要排在前面。所以先把所有入度是0的节点入队。然后每次迭代我们找出队头的节点u,所有儿子入度减1,若deg[son]是0,将son入队,完了再删除与u相连的所有边。直到队列为空,入队的顺序就是拓扑排序的顺序。本质上就是一种广度优先搜索(BFS)。