本期任務:介紹leetcode中樹的幾個經典問題的遞歸解法
【算法】【遞歸篇】【樹】第1節:leetcode 100. 相同的樹
【算法】【遞歸篇】【樹】第2節:leetcode 105. 從前序與中序遍歷序列構造二叉樹
【算法】【遞歸篇】【樹】第3節:leetcode 210. 課程表 II
【算法】【遞歸篇】【樹】第4節:leetcode 236. 二叉樹的最近公共祖先
【算法】【遞歸篇】【樹】第5節:leetcode 572. 另一個樹的子樹
【算法】【遞歸篇】【樹】第6節:leetcode 101. 對稱二叉樹
【算法】【遞歸篇】【樹】第7節:leetcode 98. 驗證二叉搜索樹
【算法】【遞歸篇】【樹】第8節:leetcode 102. 二叉樹的層序遍歷
問題來源
102. 二叉樹的層序遍歷
給你一個二叉樹,請你返回其按 層序遍歷 得到的節點值。 (即逐層地,從左到右訪問所有節點)。
示例:
二叉樹:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其層次遍歷結果:
[
[3],
[9,20],
[15,7]
]
大佬解析
代碼
"""
需求:
給你一個二叉樹,請你返回其按 層序遍歷 得到的節點值。 (即逐層地,從左到右訪問所有節點)。
思路1:
非遞歸法(BFS):使用隊列存儲每一層的節點,按層數依次出隊,若節點的左右孩子存在,則孩子入隊
思路2:
遞歸解法(DFS):先序遍歷,加入表示當前層數的index,聯想回溯思想進行類比
"""
# Definition for a binary tree node.
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
def levelOrder(self, root: TreeNode):
"""遞歸解法(DFS):先序遍歷,加入表示當前層數的index,聯想回溯思想進行類比"""
if not root:
return []
ans = list()
def helper(index, node):
if len(ans) <= index:
ans.append([])
# 先序遍歷思路
ans[index].append(node.val)
if node.left:
helper(index + 1, node.left)
if node.right:
helper(index + 1, node.right)
helper(0, root)
return ans
def levelOrder1(self, root: TreeNode):
# 非遞歸法(BFS):使用隊列存儲每一層的節點,按層數依次出隊,若節點的左右孩子存在,則孩子入隊
if not root:
return []
ans = list()
queue = [root]
while queue:
# 獲取當前隊列的長度,這個長度相當於 當前這一層的節點個數
size = len(queue)
res = list()
for i in range(size):
# 將隊列中的元素都拿出來(也就是獲取這一層的節點),放到臨時list中
# 如果節點的左/右子樹不爲空,也放入隊列中
node = queue.pop(0)
res.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
ans.append(res) # 將臨時list加入最終返回結果中
return ans