【题目】
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:2
示例 2:
输入:root = [2,null,3,null,4,null,5,null,6]
输出:5
提示:
树中节点数的范围在 [0, 105] 内
-1000 <= Node.val <= 1000
【代码】
class Solution:
def minDepth(self, root: TreeNode) -> int:
if not root:
return 0
min_depth=10**5+1
depth=1
queue=[root]
while queue:
sz=len(queue)
for i in range(sz):
node=queue.pop(0)
if not node.left and not node.right:
min_depth=min(min_depth,depth)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
depth+=1
return min_depth
【写法2】
class Solution:
def minDepth(self, root: TreeNode) -> int:
if not root:
return 0
depth=1
queue=[root]
while queue:
sz=len(queue)
for i in range(sz):
node=queue.pop(0)
if not node.left and not node.right:
return depth
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
depth+=1
return depth
【写法3-深度优先】
class Solution:
def __init__(self):
self.depth=10**5+1
def minDepth(self, root: TreeNode) -> int:
if root and not root.left and not root.right:
return 1
elif not root:
return 0
min_depth =10**6
if root.left:
min_depth=min(self.minDepth(root.left),min_depth)
if root.right:
min_depth=min(self.minDepth(root.right),min_depth)
return min_depth+1
【写法4-广度优先(推荐)】
class Solution:
def __init__(self):
self.depth=10**5+1
def minDepth(self, root: TreeNode) -> int:
if not root:
return 0
que = collections.deque([(root, 1)])
while que:
node, depth = que.popleft()
if not node.left and not node.right:
return depth
if node.left:
que.append((node.left, depth + 1))
if node.right:
que.append((node.right, depth + 1))
return 0