LeetCode 102. 二叉樹的層序遍歷
阿新 • • 發佈:2020-07-19
給你一個二叉樹,請你返回其按 層序遍歷 得到的節點值。 (即逐層地,從左到右訪問所有節點)。 示例: 二叉樹:[3,9,20,null,null,15,7], 3 / \ 9 20 / \ 15 7 返回其層次遍歷結果: [ [3], [9,20], [15,7] ] class Solution(object): def levelOrder(self, root): """ :type root: TreeNode :rtype: List[List[int]] """ if not root: return [] res = [] queue = [root] while queue: # 獲取當前佇列的長度,這個長度相當於 當前這一層的節點個數 size = len(queue) tmp = [] # 將佇列中的元素都拿出來(也就是獲取這一層的節點),放到臨時list中 # 如果節點的左/右子樹不為空,也放入佇列中 for _ in xrange(size): r = queue.pop(0) tmp.append(r.val) if r.left: queue.append(r.left) if r.right: queue.append(r.right) # 將臨時list加入最終返回結果中 res.append(tmp) return res class Solution(object): def levelOrder(self, root): """ :type root: TreeNode :rtype: List[List[int]] """ if not root: return [] res = [] def dfs(index,r): # 假設res是[ [1],[2,3] ], index是3,就再插入一個空list放到res中 if len(res)<index: res.append([]) # 將當前節點的值加入到res中,index代表當前層,假設index是3,節點值是99 # res是[ [1],[2,3] [4] ],加入後res就變為 [ [1],[2,3] [4,99] ] res[index-1].append(r.val) # 遞迴的處理左子樹,右子樹,同時將層數index+1 if r.left: dfs(index+1,r.left) if r.right: dfs(index+1,r.right) dfs(1,root) return res