对于一棵多叉树,我们可以通过 “左孩子右兄弟” 表示法,将其转化成一棵二叉树。
如果我们认为每个结点的子结点是无序的,那么得到的二叉树可能不唯一。
换句话说,每个结点可以选任意子结点作为左孩子,并按任意顺序连接右兄弟。
给定一棵包含 NN 个结点的多叉树,结点从 11 至 NN 编号,其中 11 号结点是根,每个结点的父结点的编号比自己的编号小。
请你计算其通过 “左孩子右兄弟” 表示法转化成的二叉树,高度最高是多少。
注:只有根结点这一个结点的树高度为 00。
例如如下的多叉树:
可能有以下 33 种 (这里只列出 33 种,并不是全部) 不同的 “左孩子右兄弟”表示:
其中最后一种高度最高,为 44。
输入格式
输入的第一行包含一个整数 NN。
以下 N−1N−1 行,每行包含一个整数,依次表示 22 至 NN 号结点的父结点编号。
输出格式
输出一个整数表示答案。
数据范围
对于 30%30% 的评测用例,1≤N≤201≤N≤20;
对于所有评测用例,1≤N≤1051≤N≤105。
输入样例:
5
1
1
1
2
输出样例:
4
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010,M=200010;
int h[N],e[M],ne[M],idx;
int n;
void add(int a,int b)
{
e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u)
{
int hmax=0,cnt=0;
for(int i=h[u];i!=-1;i=ne[i])
{
int j=e[i];
hmax=max(hmax,dfs(j));
cnt++;
}
return cnt+hmax;
}
int main()
{
cin>>n;
memset(h, -1, sizeof h);
for(int i=2;i<=n;i++)
{
int p;
cin>>p;
add(p,i);
}
cout<<dfs(1)<<endl;
return 0;
}