【Leetcode】拓扑排序

模版：有向图的拓扑排列

#include <iostream>
#include <cstring> //memset
using namespace std;
const int N = 1e5 + 10;

int n, m;

// 记录每个节点的入度
int d[N];

// 邻接表头h
int h[N], e[N], ne[N], idx; 
// ne是next指针 e存的节点值 idx是节点编号 头插法

// 邻接表添加有向边
void add_edge(int a, int b) {
   
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx ++ ;
}

// 模拟队列
int q[N], hh, tt = -1;

// 拓扑排序，如果存在拓扑序返回真
bool topo_sort() {
   
    // 先将所有入度为0的节点入队
    for (int i = 1; i <= n; i ++ ) {
   
        if (!d[i])
            q[ ++ tt] = i;
    }
    // BFS访问过程
    while (hh <= tt) {
   
        // 取队头节点t
        int t = q[hh ++ ];
        // 删除节点t，所以将节点t的所有后继j的入度-1
        for (int i = h[t]; i != -1; i = ne[i]) {
   
            int j = e[i];
            d[j] -- ;
            // 如果已经减少到0了，就加到队列里
            if (!d[j]) q[ ++ tt] = j;
        }
    }
    // 如果所有n个节点都加入过队列，就存在拓扑序
    return tt == n - 1;
}

int main() {
   
    // 初始化邻接表
    memset(h, -1, sizeof h);
    // 读取m条有向边，同时维护节点的入度
    cin >> n >> m;
    for (int i = 0; i < m; i ++ ) {
   
        int a, b;
        cin >> a >> b;
        add_edge(a, b);
        d[b] ++ ;
    }
    // 如果存在拓扑序，模拟队列中依次入队的顺序就是拓扑序
    if (topo_sort()) {
   
        for (int i = 0; i < n; i ++ )
            cout << q[i] << ' '