L2-026 小字辈

L2-026 小字辈

题目链接

题意

思路

1. 利用并查集找出最小的辈分层数，再判断该人的辈分层数是否为最小的辈分层数，再将倍数最小的辈分的人的编号存入一个数组中，再将其输出。

坑点

1.最后要注意一下输出，最后一个输出不带空格；

代码

#include<bits/stdc++.h>
#include <iostream>
using namespace std;
int fa[100001] = {-1};
int h[100001];//辈分，即该结点到老祖宗的高度 
int xiao[100001];
int find(int x)//利用并查集查找； 
{
	if(fa[x]==-1)
	{
		h[x]=1;//最大辈分1 
	}
	if(h[x]==0)
	{
		h[x]=find(fa[x])+1;//往后加辈分 
	}
	return h[x];
}
int main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)//初始化； 
	{
		cin>>fa[i];
	}
	for(int i=1;i<=n;i++)//找父亲节点； 
	{
		find(i);
	}
	int maxx=-1;//定义一个最大值，找辈分最小的层数和编号； 
	for(int i=1;i<=n;i++)
	{
		maxx=max(h[i],maxx);//找出最小的辈分层数 
	}
	cout<<maxx<<endl;//输出最小的辈分层数 
	int x=0;
	for(int i=1;i<=n;i++)
	{
		if(maxx==h[i])//如果这个人的辈分层数等于最小的辈分层数； 
		{
			xiao[x]=i;//就存入到xiao数组中； 
			x++;
		}
	}
	for(int i=0;i<x;i++)
	{
		cout<<xiao[i];
		if(i!=x-1)//判断一下输出情况，最后一个输出没有空格； 
		{
			cout<<" ";
		}
	}
	return 0;
}