题目
给定一个二叉树
root
,返回其最大深度。二叉树的 最大深度 是指从根节点到最远叶子节点的最长路径上的节点数。
解题
# 定义二叉树节点的类
class TreeNode:def __init__(self, val=0, left=None, right=None):self.val = valself.left = leftself.right = right# 计算二叉树最大深度的函数
def maxDepth(root: TreeNode) -> int:# 如果当前节点为空,则深度为0if not root:return 0# 递归计算左子树和右子树的深度left_depth = maxDepth(root.left)right_depth = maxDepth(root.right)# 返回左子树和右子树深度的较大者 + 1return max(left_depth, right_depth) + 1# 创建二叉树
root = TreeNode(3)
root.left = TreeNode(9)
root.right = TreeNode(20, TreeNode(15), TreeNode(7))# 计算最大深度
depth = maxDepth(root)
print(depth) # 输出 3