python leetcode刷题（26）：104. 二叉树的最大深度

最新推荐文章于 2024-03-17 20:13:00 发布

原创最新推荐文章于 2024-03-17 20:13:00 发布 · 208 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode

python刷题专栏收录该内容

53 篇文章

订阅专栏

本文探讨了计算二叉树最大深度的两种方法：深度优先搜索的递归解法和迭代解法。递归方法通过不断调用自身直到达到叶子节点，而迭代方法则使用栈来跟踪每个节点的深度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

在这里插入图片描述

解题过程：

官方答案有两个，首先是深度优先搜索，使用递归的方法：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        if not root:
            return 0
        else:
            left_height=self.maxDepth(root.left)
            right_height=self.maxDepth(root.right)
            return max(left_height,right_height)+1

递归时的发现，递归时要有if判断递归条件，以判断何时递归结束；递归本身语句意义不大可以忽略；重要的是递归后面的语句，这是每次递归都会改变的变量。
第二个答案使用的是迭代的方法：

class Solution:
    def maxDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """ 
        stack = []
        if root is not None:
            stack.append((1, root))
        
        depth = 0
        while stack != []:
            current_depth, root = stack.pop()
            if root is not None:
                depth = max(depth, current_depth)
                stack.append((current_depth + 1, root.left))
                stack.append((current_depth + 1, root.right))
        
        return depth

我们从包含根节点且深度为1的栈开始，然后继续迭代，将当前节点弹出栈并推入子节点，且每一步都会更新深度。
总结：
None