欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

拓扑排序核心代码

程序员文章站 2024-03-19 12:08:10
...

这里使用c[maxn]记录当前结点是否访问过,t为c数组当前项的下标;topo数组记录拓扑排序,n为结点数(变量数), 图G记录所有边的关系。

c[u] = 0表示从来没有访问过(即没调用过dfs[u]);c[u] = 1表示已经访问过,并且其所有的子孙也被访问过。c[u] = -1表示正在访问(即递归调用dfs[u]正在栈帧中,尚未返回)

int c[maxn], t;
int topo[maxn];
int n, G[n][n];

bool dfs(int u){
    c[u] = -1;
    for(int v = 0; v < n; v++){
        if(G[u][v]){
            if(c[v]<0) return false;
            else if (!c[v] && !dfs(v)) {
                return false;
            }
        }
    }
    c[u] = 1;topo[--t] = u;
    return true;
}

bool toposort(){
    t = n;
    memset(c, 0, sizeof(c));
    for (int u = 0; u < n; u++) {
        if(!c[u]){
            if(!dfs(u)) return false;
        }
    }
    return true;
}