题目:
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
思路:
与104求二叉树的最大深度相似,但需要注意的是,本题要求的最小深度不是任意节点的最小深度,而是叶子节点到根节点,也就是说如果一颗二叉树根节点左边为None,右边有值,此时的最小深度可不是1,而是需要在右边去找叶子结点,满足最小深度的要求。
因此,可以再104的代码上进行修改,对根基诶单的左右子节点情况进行判断。
代码:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def minDepth(self, root: Optional[TreeNode]) -> int:
def get_height(node):
if not node:
return 0
left_height = get_height(node.left)
right_height = get_height(node.right)
if node.left == None and node.right != None:
return 1 + right_height # 左边是空,返回的是右边的最小深度
if node.right == None and node.left != None:
return 1 + left_height # 右边是空,返回左边的最小深度
else:
return 1+min(left_height, right_height)
# 两边都是空或者都不为空 则返回小的值
return get_height(root)