蓝桥杯 省赛 左hz右兄弟 Python （90）

左孩子右兄弟
时间限制：1.0s内存限制：256.0MB本题总分：20分

【问题描述】

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

如果我们认为每个结点的子结点是无序的，那么得到的二叉树可能不唯一。换句话说，每个结点可以选任意子结点作为左孩子，并按任意顺序连接右兄弟。

给定一棵包含N个结点的多叉树，结点从1至N编号，其中1号结点是根，每个结点的父结点的编号比自己的编号小。请你计算其通过“左孩子右兄弟”表示法转化成的二叉树，高度最高是多少。注：只有根结点这一个结点的树高度为0。

可能有以下3种(这里只列出3种，并不是全部)不同的“左孩子右兄弟”表示：

 

其中最后一种高度最高，为4

【输入格式】

输入的第一行包含一个整数N。

以下N-1行，每行包含一个整数，依次表示2至N号结点的父结点编号。

【输出格式】

输出一个整数表示答案。

【样例输入】

5
1
1
1
2

【样例输出】

4

---

递归还是不太会用啊，卡了好久

不知道为什么最后错了一个测试点 

dic={}
n=int(input())
for i in range(2,n+1):
    a=int(input())
    if not dic.get(a):
        dic[a]=[]
    #if not dic.get(i):
        #dic[i]=[]
    dic[a].append(i)
    #dic[i].append(a)

def search(n):
    global length
    if not dic.get(n):
        dic[n]=[]
    list1=dic[n]
    #print(list1)
    length=0
    for i in list1:
        #print(length,1)
        length=max(length,search(i)+len(list1))
        #print(length,2)
    return length

    


search(1)
print(length)