拓扑排序-STL版

拓扑排序-STL版

题目描述
已知有向图，顶点从0开始编号，求它的求拓扑有序序列。
拓扑排序算法：给出有向图邻接矩阵
1.逐列扫描矩阵，找出入度为0且编号最小的顶点v
2.输出v，并标识v已访问
3.把矩阵第v行全清0
重复上述步骤，直到所有顶点输出为止

输入
第一行输入一个整数t，表示有t个有向图
第二行输入n，表示图有n个顶点
第三行起，输入n行整数，表示图对应的邻接矩阵
以此类推输入下一个图的顶点数和邻接矩阵

输出
每行输出一个图的拓扑有序序列

样例输入

2
5
0 1 0 1 1
0 0 1 0 0
0 0 0 0 1
0 0 1 0 0
0 0 0 0 0
7
0 0 0 0 0 0 0
1 0 1 1 0 0 0
1 0 0 0 0 0 0
1 0 1 0 0 0 0
0 0 0 0 0 1 1
0 1 0 0 0 0 0
0 0 0 1 0 1 0

样例输出

0 1 3 2 4 
4 6 5 1 3 2 0 

#include <iostream>
#include <queue>
using namespace std;
#define N 100

// 出度入度 状态数组
int t, n, map[N][N], in[N], start, sta[N];
queue <int> q;
vector<int> ans;

void topologySort() {
    // 拓扑排序
    q.push(start);
    while (!q.empty()) {
        int tem = q.front();
        if (sta[tem]) {
            // 重复加入了节点，所以这里多加一个判断，已经访问过的直接continue
            q.pop(); continue;
        }
        q.pop();
        sta[tem] = 1; // 标记
        ans.push_back(tem);
        for (int i = 0; i < n; i++) {
            if (sta[i]) continue; // 访问过的点
            // 能进到这个循环的点一定是入度为0的点，所以我们需要寻找它出边到达的点
            if (map[tem][i]) --in[i]; // 所到达点入度-1
            if (!in[i]) q.push(i);
        }
        
    }
}

void init() {
    for (int i = 0; i < n; i++) {
        sta[i] = 0, in[i] = 0;
    }
    ans.clear();
}

int main() {
    cin >> t;
    while (t--) {
        init();
        cin >> n;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                cin >> map[i][j];
                if (map[i][j]) in[j]++;
            }               
        }
        for (int i = 0; i < n; i++) {
            if (!in[i]) {
                start = i;
                break;
            }
        }
        topologySort();
        for (int i = 0; i < ans.size(); i++) {
            cout << ans[i] << " ";
        }
        cout << endl; 
    }
    return 0;
}