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

大致思路

代码实现

#include<bits/stdc++.h>
using namespace std;
const int N=520;
vector<int>p[N];   //vector变长数组来写邻接表 
queue<int>q;       //队列操作 
int d[N];          //统计入度 
int n,x,cnt,ans[N];  //ans数组记录答案
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		while(cin>>x,x){
			p[i].push_back(x);    //构造邻接表 
			d[x]++;         //记录x的入度 
		}
	}
	for(int i=1;i<=n;i++)if(!d[i])q.push(i);   //寻找入度为0的点
	
	while(q.size()){
		int t=q.front();
		q.pop();
		ans[cnt++]=t;
		for(int i=0;i<p[t].size();i++){
			d[p[t][i]]--;
			if(d[p[t][i]]==0)q.push(p[t][i]);   //如果删除边后入度为0，就放入队列 
		}
	} 
	for(int i=0;i<cnt;i++)cout<<ans[i]<<" ";
	cout<<endl;
	return 0;
}