系列索引:菜菜的刷题日记 | 被LeetCode用Python狂虐的那段日子
考前抱佛脚!
【题目】
题目链接:http://lx.lanqiao.cn/problem.page?gpid=T2898
【我的代码】
我的思路是:
前面一段是输入处理,这里我用了字典来存储每一个元素的孩子,大概效果如下图。然后就是一个递归
,主要是查询每个元素的孩子个数以及孩子的孩子的个数的最大值。(可能有点绕,不过递归就是如此)
N = int(input())
d = dict({i: [] for i in range(1, N+1)})
for i in range(2, N+1):
x = int(input())
d[x].append(i)
# print(d)
def find(x): #参数为要查询的节点标号,这里直接使用1,从头开始
length = len(d[x])
if d[x]:
m = []
for i in d[x]:
m.append(find(i))
return length + max(m)
else:
return 0
print(find(1))
对字典操作不熟悉的可以看这篇:
菜菜的Python学习日记 | 集合与字典(定义、创建、增删、运算)详解
【思考】
拿个省三就行!
欢迎热爱Python的同学关注我,后期计划开一个Python大作业专栏!
Python力扣题解系列持续更新,欢迎
点赞收藏
+关注
上一篇:菜菜的刷题日记 | 蓝桥杯 — 十六进制转八进制(纯手撕版)附进制转换笔记
下一篇:菜菜的刷题日记 | 一行代码解决“存在重复元素“
本人水平有限,文章中不足之处欢迎下方👇评论区批评指正~如果感觉对你有帮助,点个赞👍 支持一下吧 ~
不定期分享 有趣、有料、有营养内容,欢迎 订阅关注 🤝 我的博客 ,期待在这与你相遇 ~