拓扑排序

#include<iostream>
#include<vector>
#include<utility>
#include<algorithm>
#include<stack>
#include<cassert>
using namespace std;
enum Color{white,gray,black};

vector<pair<int,int>> toplogical_list;

class Node{
public:
    int index;
    Node* next=nullptr;//
    Node(int i):index(i){}
};

class VNode{
public:
    char vertex;
    int dist;
    int final;
    int indegree;
    Color color=white;
    int prev=-1;
    Node* firstNode=nullptr;
    VNode(char c):vertex(c){}
};


typedef struct Graph{
    int EdgeNum;
    vector<VNode> Adj;
}Graph;

void DFS_VISIT(Graph &G,int u);
int timex;
void DFS(Graph &G){
    for(int i=0;i<G.Adj.size();i++){
        G.Adj[i].color = white;
        G.Adj[i].prev = -1;
    }
     timex =0;
     //begin from a certain vertex
     //DFS_VISIT(G,inndex);
    for(int i=0;i<G.Adj.size();i++){
        if(G.Adj[i].color == white){
            DFS_VISIT(G,i);
        }
    }
}

void DFS_VISIT(Graph &G,int u){
    timex += 1;
    G.Adj[u].dist = timex;
    G.Adj[u].color = gray;
    //cout<<G.Adj[u].vertex<<",";
    
    Node* pv = G.Adj[u].firstNode;
    while(pv != nullptr){
        int v= pv->index;
        if(G.Adj[v].color == white){
            G.Adj[v].prev = u;
            DFS_VISIT(G,v);
        }
        pv = pv->next;
    }
    G.Adj[u].color = black;
    timex += 1;
    G.Adj[u].final = timex;
    toplogical_list.push_back(make_pair(u,timex));
}

int comp(pair<int,int> a,pair<int,int> b){
    if(a.second>b.second){
        return -1;
    }
    else if(a.second==b.second){
        return 0;
    }
    else{
        return 1;
    }
}
void toplogical_sort(Graph G){
    DFS(G);
    sort(toplogical_list.begin(),toplogical_list.end(),comp);
    for(const auto &c:toplogical_list ){
        cout<<G.Adj[c.first].vertex<<",";
    }
    cout<<endl;
}

void find_indegree(Graph G,vector<int> &indegree){
    for(int i=0;i<G.Adj.size();++i){
        indegree.push_back(G.Adj[i].indegree);
    }
}
void toplogical_sort2(Graph G){
    vector<int> indegree;
    find_indegree(G,indegree);
    stack<int> S;
    for(int i=0;i<G.Adj.size();++i){
        if(!indegree[i]){
            S.push(i);
        }
    }
    int count = 0;
    while(!S.empty()){
        int i=S.top();
        S.pop();
        cout<<G.Adj[i].vertex<<",";
        ++count;
        for(auto p=G.Adj[i].firstNode;p != nullptr;p = p->next){
            int k=p->index;
            if(!(--indegree[k])){
                S.push(k);
            }
        }
    }
    assert(count == G.Adj.size());
}

int main(){
    Graph G;
    G.EdgeNum = 9;
    //vector<char> v{‘v‘,‘r‘,‘s‘,‘w‘,‘x‘,‘y‘,‘t‘,‘u‘};
    vector<char> v{‘v‘,‘r‘,‘s‘,‘w‘,‘x‘,‘y‘,‘t‘,‘u‘,‘p‘};
    for(int i=0;i<v.size();i++){
        G.Adj.push_back(VNode(v[i]));
    }

    int i,j;
    for(int k=0;k<G.EdgeNum;k++){
        cin>>i>>j;
        Node* p = new Node(j);
        p->next = G.Adj[i].firstNode;
        G.Adj[i].firstNode = p;

        ++G.Adj[j].indegree;
        /*
        p = new Node(i);
        p->next = G.Adj[j].firstNode;
        G.Adj[j].firstNode = p;
        */
    }
    //DFS(G);
    toplogical_sort(G);
    toplogical_sort2(G);
    cout<<endl;
    return 0;
}