Python版-LeetCode 学习:111. 二叉树的最小深度

给定一个二叉树,找出其最小深度。

最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

说明: 叶子节点是指没有子节点的节点。

示例:给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回它的最小深度  2.

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-depth-of-binary-tree

方法1:BFS算法,从root开始遍历每一层的节点,如果出现节点的左右子节点为空的情况,则还回此时的深度

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

class Solution:
    def minDepth(self, root: TreeNode) -> int:
        if  not root: return 0

        depth=1

        queue=[]
        queue.append(root)
        
        while queue !=[]:
            for i in range(len(queue)):
                # 获取进入队列的第一个节点
                cur = queue.pop(0)
                # 判断左右子树是否为空
                if (cur.left is None) and (cur.right is None):
                    return depth
                # 判断左右子树不为空时,加入队列
                if cur.left is not None:
                    queue.append(cur.left)
                if cur.right is not None:
                    queue.append(cur.right)
            depth+=1
        
        return depth

方法2:使用递归算法,该算法是遍历了root的左右子树,判断左右子树中哪个深度最小。中间不需要暂存队列来遍历,相对方法1代码简单,效率也更快。

class Solution:
    def minDepth(self, root: TreeNode) -> int:
        if not root: return 0
        # 左子树不为空,遍历右子树,深度加1
        if not root.left: return 1 + self.minDepth(root.right)
        # 右子树不为空,遍历左子树,深度加1
        if not root.right: return 1 + self.minDepth(root.left)
        # 还回左右子树总深度最小的值
        return 1 + min(self.minDepth(root.left), self.minDepth(root.right))

猜你喜欢

转载自blog.csdn.net/guyu1003/article/details/107289233