例题6-15 UVA10305 Ordering Tasks（30行AC代码）

紫书刷题进行中，题解系列【GitHub|CSDN】

例题6-15 UVA10305 Ordering Tasks（30行AC代码）

题目大意

给出有向无环图，输出拓扑排序（若有多个则任意输出一个）

思路分析

定义邻接表adj存储有向无环图，定义indegree记录每个点的入度

vector<vector<int> > adj(n+1); // 邻接表
vector<int> indegree(n+1), ans; // 入度表，从1开始；存储结果

定义队列q模拟拓扑排序过程，具体过程如下：

• 初始化：将入度为0的顶点加入队列q，并置入度indegree为-1，表示已访问
• 若队列非空，队首u出队（出队即可输出），将u能到达的下一个顶点的入度减1，再将入度为0的顶点入队

注意点

• m可能等于0，n不为0，如n=5,m=0，未考虑会WA

AC代码（C++11，拓扑排序）

#include<bits/stdc++.h>
using namespace std;
int n, m, a, b;
void update(vector<int>& indegree, queue<int>& q) { // 更新队列
    for (int i = 1; i <= n; i ++) {
        if (indegree[i] == 0) { // 入度为0
            q.push(i); // 入队
            indegree[i] = -1; // 标记为已访问
        }
    }
}
int main() {
    while (cin >>n >>m && (n != 0 || m != 0)) { // 仅m=0是合法的，如n=5,m=0
        vector<vector<int> > adj(n+1); // 邻接矩阵
        vector<int> indegree(n+1), ans; // 入度表，从1开始；存储结果
        while (m --) {
            cin >>a >>b;
            adj[a].push_back(b);
            indegree[b] ++; // 入度累加
        }
        queue<int> q; update(indegree, q); // 初始化
        while (!q.empty()) {
            int u = q.front(); q.pop(); ans.push_back(u);
            for (int v : adj[u]) indegree[v] --; // 更新入度表
            update(indegree, q); // 更新队列
        }
        for (int i = 0; i < ans.size(); i ++) printf("%d%s", ans[i], i == ans.size()-1 ? "\n" : " ");
    }
    return 0;
}