PTA天梯 L2-1 小字辈（左子右兄加强版）

题目描述

本题给定一个庞大家族的家谱，要请你给出最小一辈的名单。
需要注意的是，家族成员的输入方式，是按照“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

思路

先找到根节点（可以标记每个人的兄弟和儿子，因为只用一个老祖宗，最后没被标记的就是老祖宗），然后直接从根节点dfs搜一遍，可以每次更新最大深度，同时把此时最大深度的人存下来（vector）
（这个题就是PTA天梯练习集中L2-026-小字辈的简单变形，选拔赛的时候没做出来属实可惜（当时没刷到“原题”也是很可惜，当然还是刷的少，属实菜了），可能也是做的太慢了，最后没时间改这个题了，比赛完再看看当时还是可以做的，还是得多练啊，啥时候才能像ybl大佬一样强啊/(ㄒoㄒ)/~~）

代码1

#include<iostream>
#include<vector>
#include<algorithm>
#define pii pair<int, int>
using namespace std;
const int maxn=1e5+10;
vector<pii>p[maxn];
vector<int>ans;
bool vis[maxn];
int maxx;
void dfs(int pos,int t)
{
	if(t>maxx)
    {
        maxx=t;
        ans.clear();
        ans.push_back(pos);
    }
    else if(maxx==t) ans.push_back(pos);
    for(int i=0;i<p[pos].size();i++)
    {
    	if(p[pos][i].first==0&&p[pos][i].second==0) continue;
    	if(p[pos][i].first) dfs(p[pos][i].first,t+1);//儿子，辈分 + 1 
    	if(p[pos][i].second) dfs(p[pos][i].second,t);//兄弟，辈分不加 
	}
}
int main()
{
    int n,pos;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
    	int a,b,c;
        cin>>a>>b>>c;
        vis[b]=1,vis[c]=1;
        p[a].push_back(make_pair(b,c));
    }
    for(int i=1;i<=n;i++)
    {
    	if(vis[i]==0)
    	{
    		pos=i;
    		break;
		} 
	}
    dfs(pos,1);
    sort(ans.begin(),ans.end());
    cout<<maxx<<endl;
    for(vector<int>::iterator it=ans.begin();it!=ans.end();it++)
    {
    	if(it==ans.begin()) cout<<*it;
    	else cout<<' '<<*it;
	}
    return 0;
}

还可以开个数组记录每个节点的深度（辈分）（最后深度跟最大深度相同的人输出）

代码2

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long llu;
const int maxn=1e5+10;
struct node
{
	int so=0;
	int br=0;
}p[maxn];
bool vis[maxn];
int deep[maxn],ans[maxn],tot;
void dfs(int pos)
{
	if(!p[pos].br&&!p[pos].so) return ;
	if(p[pos].br)
	{
		deep[p[pos].br]=deep[pos];
		dfs(p[pos].br);
	}
	if(p[pos].so)
	{
		deep[p[pos].so]=deep[pos]+1;
		dfs(p[pos].so);
	}
}
int main( )
{
	int n,pos;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		int a,b,c;
		cin>>a>>b>>c;
		p[a].so=b,p[a].br=c;
		vis[p[a].so]=1,vis[p[a].br]=1;
	}
	for(int i=1;i<=n;i++)
	{
		if(vis[i]==0)
		{
			pos=i;
			break;
		}
	}
	deep[pos]=1;
	dfs(pos);
	int maxx=0;
	for(int i=1;i<=n;i++)
	{
		if(maxx<deep[i]) 
		{
			maxx=deep[i];
		}
	}
	cout<<maxx<<endl;
	for(int i=1;i<=n;i++)
	{
		if(deep[i]==maxx)
		{
			ans[tot++]=i;
		}
	}
	sort(ans,ans+tot);
	for(int i=0;i<tot;i++)
	{
		if(i==0) cout<<ans[i];
		else cout<<' '<<ans[i];
	}
	return 0;
}