给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
示例:给定二叉树 [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))