LeetCode(Python)—— 二叉树的最大深度(简单)

二叉树的最大深度

概述:给定一个二叉树,找出其最大深度。二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。叶子节点是指没有子节点的节点。

给定二叉树 [3,9,20,null,null,15,7]
    3
   / \
  9  20
    /  \
   15   7
返回它的最大深度 3 

方法一:深度优先搜索(DFS)

思路:在计算当前二叉树的最大深度时,可以先递归计算出其左子树和右子树的最大深度,递归在访问到空节点时退出。

# 深度优先搜索(DFS)
class Solution:
    def maxDepth(self, root: Optional[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

方法二:广度优先搜索(BFS)

思路:我们需要将队列里的所有节点都拿出来进行拓展,这样能保证每次拓展完的时候队列里存放的是当前层的所有节点,即我们是一层一层地进行拓展。

# 广度优先搜索(BFS)
class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        if not root: # 空树
            return 0
        queue = [root]
        height = 0
        while queue:
            n = len(queue)
            for i in range(n):
                node = queue.pop(0)
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            height += 1
        return height

总结

一看就会,写就不对,递归一生之敌。

你可能感兴趣的:(leetcode,算法,数据结构,python,深度优先)