原題
給定一個(gè)二叉樹(shù),找出其最小深度。
二叉樹(shù)的最小深度為根節(jié)點(diǎn)到最近葉子節(jié)點(diǎn)的距離。
給出一棵如下的二叉樹(shù):
1
/ \
2 3
/ \
4 5
這個(gè)二叉樹(shù)的最小深度為 2
解題思路
- 一定要注意定義,比如下面這棵樹(shù)的最小深度是3不是1
1
\
2
\
3
- 方法和找最大深度一樣,Divide & Conquer
只需要判斷如果左右兒子中又None,則返回非None的兒子的高度+1
完整代碼
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def minDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if not root:
return 0
if None in [root.left, root.right]:
return max(self.minDepth(root.left), self.minDepth(root.right)) + 1
else:
return min(self.minDepth(root.left), self.minDepth(root.right)) + 1