111.二叉树的最小深度
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
题解
# 层次遍历求解(BFS)
class Solution:
def minDepth(self, root: Optional[TreeNode]) -> int:
if not root:
return 0
queue = [root]
deep = 0
now_node = 1
next_node = 0
while len(queue) > 0:
n = queue.pop(0)
if n.left:
next_node += 1
queue.append(n.left)
if n.right:
next_node += 1
queue.append(n.right)
if (not n.left) and (not n.right):
return deep + 1
now_node -= 1
if now_node == 0:
now_node = next_node
next_node = 0
deep += 1
# 递归求解(DFS)
class Solution:
def minDepth(self, root: Optional[TreeNode]) -> int:
def deep(p):
if not p:
return 0
deep_left = deep(p.left)
deep_right = deep(p.right)
md = min(deep_left, deep_right)
if md == 0:
return max(deep_left, deep_right) + 1
else:
return md + 1
d = deep(root)
return d
思路
层次遍历(BFS)求解:目的是找到深度最小的叶子,用层次遍历树时,第一个找到的叶子的深度就是答案。
递归(DFS)求最下深度。