1. 问题描述:
给定一个 N 叉树,找到其最大深度。最大深度是指从根节点到最远叶子节点的最长路径上的节点总数。N 叉树输入按层序遍历序列化表示,每组子节点由空值分隔(请参见示例)。
示例 1:
输入:root = [1,null,3,2,4,null,5,6]
输出:3
示例 2:
输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出:5
提示:
树的深度不会超过 1000 。
树的节点数目位于 [0, 10 ^ 4] 之间。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-depth-of-n-ary-tree
2. 思路分析:
分析题目可以知道我们递归N叉树即可,在递归的时候维护当前节点的深度,并且需要声明一个全局变量来记录当前的N叉树的最大深度res,在递归的时候更新res即可。第二种写法是求解每一个子节点的最大深度。
3. 代码如下:
class Node:
def __init__(self, val=None, children=None):
self.val = val
self.children = children
class Solution:
res = 0
def dfs(self, root: Node, d: int):
if not root: return
if d > self.res: self.res = d
for child in root.children:
self.dfs(child, d + 1)
def maxDepth(self, root: 'Node') -> int:
self.dfs(root, 1)
return self.res
class Node:
def __init__(self, val=None, children=None):
self.val = val
self.children = children
class Solution:
def dfs(self, root: Node, d: int):
if not root: return 0
res = 0
# 求解每一个子节点的最大深度
for child in root.children:
res = max(res, self.dfs(child, d + 1))
return res + 1
def maxDepth(self, root: 'Node') -> int:
return self.dfs(root, 0)