拓撲排序核心程式碼

bkjs626發表於2017-10-21

這裡使用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;
}

相關文章