LeetCode.jpg
111. 二叉樹的最小深度
給定一個(gè)二叉樹婴程，找出其最小深度。
最小深度是從根節(jié)點(diǎn)到最近葉子節(jié)點(diǎn)的最短路徑上的節(jié)點(diǎn)數(shù)量。
說明: 葉子節(jié)點(diǎn)是指沒有子節(jié)點(diǎn)的節(jié)點(diǎn)。
示例:
給定二叉樹 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最小深度 2.
Python3實(shí)現(xiàn)
DFS 深度優(yōu)先
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
# @author:leacoder
# @des: DFS 深度優(yōu)先 二叉樹的最大深度 時(shí)間復(fù)雜度 O(n)
class Solution:
def maxDepth(self, root: TreeNode) -> int:
if not root:
return 0 #入?yún)⑿ｒ?yàn)
else: # DFS
leftlevel = self.maxDepth(root.left) # 遞歸 左子樹 得到其深度
rightlevel = self.maxDepth(root.right) # 遞歸 右子樹 得到其深度
return 1 + max(leftlevel,rightlevel) # 取最大深度
BFS廣度優(yōu)先
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
# @author:leacoder
# @des: BFS 廣度優(yōu)先 二叉樹的最大深度 時(shí)間復(fù)雜度 O(n)
class Solution:
def maxDepth(self, root: TreeNode) -> int:
if not root:
return 0 #入?yún)⑿ｒ?yàn)
else: # BFS
queue = collections.deque()
queue.append(root) # 輔助隊(duì)列
maxlevel = 0 #層級(jí)記錄
while queue:
levelsize = len(queue) #
for i in range(levelsize):
node = queue.popleft()
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
maxlevel +=1 # 廣度優(yōu)先搜索 每層搜完 層級(jí)+1
return maxlevel
GitHub鏈接：
https://github.com/lichangke/LeetCode
知乎個(gè)人首頁(yè)：
https://www.zhihu.com/people/lichangke/
個(gè)人Blog:
https://lichangke.github.io/
歡迎大家來(lái)一起交流學(xué)習(xí)
