1. 程式人生 > 其它 >劍指 Offer 32 - II. 從上到下列印二叉樹 II_CodingPark程式設計公園

劍指 Offer 32 - II. 從上到下列印二叉樹 II_CodingPark程式設計公園

技術標籤:leetcodeleetcode

問題

從上到下按層列印二叉樹,同一層的節點按從左到右的順序列印,每一層列印到一行

例如:
給定二叉樹: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其層次遍歷結果:

[
  [3],
  [9,20],
  [15,7]
]
 

連結:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof

解答

class Solution:
    def levelOrder
(self, root: TreeNode) -> List[List[int]]: if not root: return [] res, queue = [], collections.deque() queue.append(root) while queue: tmp = [] for _ in range(len(queue)): node = queue.popleft() tmp.append(node.
val) if node.left: queue.append(node.left) if node.right: queue.append(node.right) res.append(tmp) return res

在這裡插入圖片描述

在這裡插入圖片描述