洛谷-B3644 【模板】拓扑排序 / 家谱树

Abstract

传送门
这篇题解主要介绍如何使用BFS去实现拓扑排序。

---

Idea

见下方注释

---

Code

#include <bits/stdc++.h>
using namespace std;

int n;                    // 记录点的数量
int in[105];              // 记录点的入度
vector<vector<int>> sons; // 记录每个点的儿子节点

int main()
{
    cin >> n;
    sons.resize(n + 1);
    for (int i = 1; i <= n; i++)
    {
        int x;
        while (1)
        {
            cin >> x;
            if (x == 0)
            {
                break;
            }
            in[x]++;
            sons[i].push_back(x);
        }
    }
    queue<int> q;
    for (int i = 1; i <= n; i++)
    {
        if (in[i] == 0) // 将所有入度为0的点添加到容器q中
        {
            q.push(i);
        }
    }
    vector<int> ans; // 记录拓扑排序后的序列
    while (!q.empty())
    {
        int u = q.front();
        q.pop();                                 // 将u节点删除
        ans.push_back(u);                        // u节点加入排好后的序列中
        for (int i = 0; i < sons[u].size(); i++) // 将u节点的所有儿子节点入度减一
        {
            in[sons[u][i]]--;
            if (in[sons[u][i]] == 0) // 若此儿子节点入度变为0，那么将他加入容器q
            {
                q.push(sons[u][i]);
            }
        }
    }
    for (int i = 0; i < ans.size(); i++) // 按顺序将序列打印出来
    {
        cout << ans[i] << " ";
    }
    return 0;
}