题干:

L2-2 小字辈(左子右兄加强版) (25 分)

本题给定一个庞大家族的家谱,要请你给出最小一辈的名单。
需要注意的是,家族成员的输入方式,是按照“Left-child Right-sibling”(左儿子 右兄弟)的格式
【敲黑板】sibling的意思是兄弟/姐妹,是亲兄弟,不是“表兄弟”!2019年春季PAT中不认识单词sibling的孩纸请牢记……
注:本题是在浙江大学陈越老师的GPLT-2018决赛L2-2题目“小字辈”题意基础上改编而成。

输入格式:

输入在第一行给出家族人口总数 N(不超过 100 000 的正整数) —— 简单起见,我们把家族成员从 1 到 N 编号。
随后的N行,每行给出1名家族成员的编号以及他的亲属关系,格式为3个以空格间隔的整数A B C
其中的A(范围1~N)是1名家族成员编号(每行的编号必定各不相同)
B的范围是0~N,当B不为0,表示A有儿子B
C的范围是0~N,当C不为0,表示A有兄弟C
题目保证输入是正确的,即:只有一个家族(没有分离成家族群),且家族中只有一位老祖宗(老祖宗即:他不是任何其他人的儿子)

输出格式:

首先输出最小的辈分(老祖宗的辈分最大、数值为 1,子孙的辈分数值逐级递增1)。
然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔,行首尾不得有多余空格。

输入样例1:

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

输出样例1:

3
4 5

输入样例2:

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

输出样例2:

3
4 5

解题报告:

  首先用入度为0的特点找到根节点,然后递归维护每个节点的深度。最后按要求输出就行了。

AC代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<cmath>
#include<cstring>
typedef long long ll;
using namespace std;
const int MAX = 2e5 + 6;
int n;
int no[MAX],son[MAX],bro[MAX];
pair<int,int> pr[MAX];
int dep[MAX];
bool bk[MAX];
int ans2[MAX];
void go(int rt) {
	if(pr[rt].first == 0 && pr[rt].second == 0) return ;
	if(pr[rt].first != 0) {
		dep[pr[rt].first] = dep[rt] + 1;
		go(pr[rt].first);
	}
	if(pr[rt].second != 0) {
		dep[pr[rt].second] = dep[rt] ;
		go(pr[rt].second);
	}
}
int main() 
{
	cin>>n;
	for(int i = 1; i<=n; i++) {
		scanf("%d%d%d",&no[i],&son[i],&bro[i]);
		pr[no[i]] = make_pair(son[i],bro[i]);
		bk[son[i]]=1;bk[bro[i]]=1;
	}
	int rt=1;
	for(int i = 1; i<=n; i++) {
		if(bk[i] == 0){
			rt = i;break;
		}
	}
	dep[rt] = 1;
	go(rt);
	int ans1 = 0;
	for(int i = 1; i<=n; i++) {
		ans1 = max(ans1,dep[i]);
	}
	int tot = 0;
	for(int i = 1; i<=n; i++) {
		if(dep[i] == ans1) {
			ans2[++tot] = i;
		}
	}
	sort(ans2+1,ans2+tot+1);
	printf("%d\n",ans1);
	for(int i = 1; i<=tot; i++) {
		printf("%d%c",ans2[i],i == tot ? '\n' : ' ');
	}
	return 0 ;
}