LeetCode 劍指offer——從上到下列印二叉樹 II、從上到下列印二叉樹 III

從上到下列印二叉樹 II

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

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

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

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

程式碼：

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        res = []
        if not root: return res 
        queue_1, queue_2 = [], []
        queue_2.append(root)
        while True:
            tmp_list = []
            while(queue_2):
                tmp = queue_2.pop(0)
                tmp_list.append(tmp.val)
                if tmp.left: queue_1.append(tmp.left)
                if tmp.right: queue_1.append(tmp.right)
            res.append(tmp_list)
            if len(queue_1) == 0: break
            queue_2 = queue_1
            queue_1 = []
        return res

利用佇列實現層序遍歷即可，此題要求同一層一起列印，用一個額外列表輔助儲存即可。時間複雜度O(n)，空間複雜度O(n)。

從上到下列印二叉樹 III

請實現一個函式按照之字形順序列印二叉樹，即第一行按照從左到右的順序列印，第二層按照從右到左的順序列印，第三行再按照從左到右的順序列印，其他行以此類推。

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

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

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

程式碼：

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        res = []
        if not root: return res 
        queue1, queue2 = [], []
        queue2.append(root)
        while True:
            tmp_list = []
            while queue2:
                tmp = queue2.pop(0)
                tmp_list.append(tmp.val)
                if tmp.left: queue1.append(tmp.left)
                if tmp.right: queue1.append(tmp.right)
            res.append(tmp_list) if len(res)%2 == 0 else res.append(tmp_list[::-1])
            if not queue1: break
            queue2 = queue1
            queue1 = []
        return res

此題和上題的區別在於要求之型列印，這裡利用了Python列表的切片翻轉方法，[::-1]，假設根節點所在層為第0層，從上到下偶數層不翻轉，奇數層翻轉。

時間複雜度O(n)，空間複雜度O(n)。