计算二叉树的深度（LeetCode）

最新推荐文章于 2024-10-05 12:29:44 发布

好好学习Py

最新推荐文章于 2024-10-05 12:29:44 发布

阅读量360

点赞数 11

分类专栏：算法与数据结构文章标签： leetcode 算法 python 数据结构

本文链接：https://blog.csdn.net/weixin_74254879/article/details/141792478

版权

算法与数据结构专栏收录该内容

82 篇文章 0 订阅

订阅专栏

题目

某公司架构以二叉树形式记录，请返回该公司的层级数。

解题

# 定义二叉树的节点
class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

def maxDepth(root):
    if not root:
        return 0
    
    # 计算左子树和右子树的深度
    left_depth = maxDepth(root.left)
    right_depth = maxDepth(root.right)
    
    # 树的深度是左子树和右子树深度的最大值加1
    return max(left_depth, right_depth) + 1

# 示例用法
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)

# 调用函数，输出结果
print(maxDepth(root))  # 输出3