Python算法——树的最大深度和最小深度

class TreeNode:
    def __init__(self, value):
        self.val = value
        self.left = None
        self.right = None

def max_depth(root):
    if not root:
        return 0
    left_depth = max_depth(root.left)
    right_depth = max_depth(root.right)
    return 1 + max(left_depth, right_depth)

def min_depth(root):
    if not root:
        return 0
    left_depth = min_depth(root.left)
    right_depth = min_depth(root.right)
    return 1 + (min(left_depth, right_depth) if left_depth and right_depth else max(left_depth, right_depth))

# 构建二叉树
"""
        1
       / \
      2   3
     / \
    4   5
"""
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
python
Copy code
# 计算最大深度和最小深度
max_depth_value = max_depth(root)
min_depth_value = min_depth(root)

print("树的最大深度:", max_depth_value)
print("树的最小深度:", min_depth_value)

树的最大深度: 3
树的最小深度: 2