题目描述:
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
示例:
1.
输入:root = [3,9,20,null,null,15,7]
输出:2
2.输入:root = [2,null,3,null,4,null,5,null,6]
输出:5
思路: 同求二叉树的最大深度一样
- 广度优先搜索——层次遍历,第一个叶子节点的深度就是所求的最小深度。
class Solution:
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
时间复杂度、空间复杂度都为O(n)。
- 深度优先搜索:递归的思想,需要注意的是,求最小的时候,不能直接min,因为一旦一个子树为空,就返回0但是该结点不是叶子节点不能返回0,否则结果会出错,具体分情况讨论,详见代码。
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def minDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if not root:
return 0
left = self.minDepth(root.left)
right = self.minDepth(root.right)
if root.left and not root.right:
return 1 + left
elif root.right and not root.left:
return 1 + right
else:
return 1 + min(left, right)
时间复杂度:O(N),其中 N是树的节点数。对每个节点访问一次。
空间复杂度:O(H),其中 H 是树的高度。空间复杂度主要取决于递归时栈空间的开销,最坏情况下,树呈现链状,空间复杂度为 O(N)。平均情况下树的高度与节点数的对数正相关,空间复杂度为 O(logN)。