第十二届蓝桥杯省赛第一场C++A/C组3422. 左孩子右兄弟

对于一棵多叉树，我们可以通过 “左孩子右兄弟” 表示法，将其转化成一棵二叉树。

如果我们认为每个结点的子结点是无序的，那么得到的二叉树可能不唯一。

换句话说，每个结点可以选任意子结点作为左孩子，并按任意顺序连接右兄弟。

给定一棵包含 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;
}