B3644 拓扑排序 & 家谱树 - 洛谷

【模板】拓扑排序 / 家谱树

题目描述

有个人的家族很大，辈分关系很混乱，请你帮整理一下这种关系。给出每个人的后代的信息。输出一个序列，使得每个人的后辈都比那个人后列出。 ## 输入格式 第 \(1\) 行一个整数 \(N\)（\(1 \le N \le 100\)），表示家族的人数。接下来 \(N\) 行，第 \(i\) 行描述第 \(i\) 个人的后代编号 \(a_{i,j}\)，表示 \(a_{i,j}\) 是 \(i\) 的后代。每行最后是 \(0\) 表示描述完毕。 ## 输出格式 输出一个序列，使得每个人的后辈都比那个人后列出。如果有多种不同的序列，输出任意一种即可。 ## 样例 #1 ### 样例输入 #1

5
0
4 5 1 0
1 0
5 3 0
3 0

### 样例输出 #1

2 4 5 3 1

很久之前可能是 Gpt 写的代码 邻接表的写法：

#include <bits/stdc++.h>
using namespace std;
vector<int> topsort(vector<vector<int>> &G)
{
    int n = G.size();
    vector<int> du(n, 0);
    for (int i = 0; i < n; i++)
        for (int j = 0; j < G[i].size(); j++)
            du[G[i][j]]++;
    queue<int> q;
    for (int i = 0; i < n; i++)
        if (!du[i])
            q.push(i);
    vector<int> result;
    while (!q.empty())
    {
        int person = q.front();
        q.pop();
        result.push_back(person);
        for (int i = 0; i < G[person].size(); i++)
        {
            int son = G[person][i];
            du[son]--;
            if (!du[son])
                q.push(son);
        }
    }
    return result;
}
int main()
{
    int n;
    cin >> n;
    vector<vector<int>> G(n);
    for (int i = 0; i < n; i++)
    {
        int son;
        while (cin >> son && son != 0)
            G[i].push_back(son - 1);
    }
    vector<int> result = topsort(G);
    for (int i = 0; i < result.size(); i++)
        cout << result[i] + 1 << ' ';
}

[[../../ACM/图论/链式前向星存图]]

#include <bits/stdc++.h>
using namespace std;
struct Edge
{
    int to, next;
} edge[2*N];
int head[N], du[N];
int cnt, n;
void addEdge(int u, int v)
{
    edge[cnt] = {v, head[u]};
    head[u] = cnt++;
}
void topsort()
{
    queue<int> q;
    for (int i = 1; i <= n; i++)
        if (!du[i])
            q.push(i);
    vector<int> result;
    while (!q.empty())
    {
        int person = q.front();
        q.pop();
        result.push_back(person);
        for (int i = head[person]; i != -1; i = edge[i].next)
        {
            int son = edge[i].to;
            du[son]--;
            if (!du[son])
                q.push(son);
        }
    }
    for (int i = 0; i < result.size(); i++)
        cout << result[i] << " ";
}
int main()
{
    cin >> n;
    memset(head, -1, sizeof head);// memset(du, 0, sizeof du);
    for (int i = 1; i <= n; i++)
    {
        int x;
        while (cin >> x && x != 0)
            addEdge(i, x), du[x]++;
    }
    topsort();
}