題目描述

從上往下打印出二叉樹的每個節點，同層節點從左至右列印。

解題思路

條件反射地想通過遞迴解決，結果硬是沒有找到合適的解決思路，也許遞迴的方式不是很適合這種型別的題目吧~
利用佇列的先進先出(FIFO)特性解決。每從佇列頭部獲取一個節點，就將該節點的左右子節點存入佇列的尾部。如此往復，直至佇列為空。這篇部落格內的圖片和表格很形象的展示了程式執行過程中佇列內值的變化過程：

二叉樹樣例：

列印二叉樹每層節點時佇列內值的變化：

Python程式碼

def PrintFromTopToBottom(self, root):
    queue = []
    result = []
    if
 
 root == None:
        return result

    queue.append(root)
    while queue:
        newNode = queue.pop(0)
        result.append(newNode.val)
        if newNode.left != None:
            queue.append(newNode.left)
        if newNode.right != None:
            queue.append(newNode.right)
    return result
 

帶測試的Python程式碼：

class Solution:
   #從上往下打印出二叉樹的每個節點，同層節點從左至右列印
    def PrintFromTopToBottom(self, root):
        array = []
        result = []
        if root == None:
            return result

        array.append(root)
        while array:
            newNode = array.pop(0)
            result.append(newNode.val)
            if
 
 newNode.left != None:
                array.append(newNode.left)
            if newNode.right != None:
                array.append(newNode.right)
        return result

    # 給定二叉樹的前序遍歷和中序遍歷，獲得該二叉樹
    def getBSTwithPreTin(self, pre, tin):
        if len(pre)==0 | len(tin)==0:
            return None

        root = treeNode(pre[0])
        for order,item in enumerate(tin):
            if root .val == item:
                root.left = self.getBSTwithPreTin(pre[1:order+1], tin[:order])
                root.right = self.getBSTwithPreTin(pre[order+1:], tin[order+1:])
                return root

class treeNode:
    def __init__(self, x):
        self.left = None
        self.right = None
        self.val = x

if __name__ == '__main__':
    flag = "printTreeNode"
    solution = Solution()
    preorder_seq = [1, 2, 4, 7, 3, 5, 6, 8]
    middleorder_seq = [4, 7, 2, 1, 5, 3, 8, 6]
    treeRoot1 = solution.getBSTwithPreTin(preorder_seq, middleorder_seq)
    if flag == "printTreeNode":
        newArray = solution.PrintFromTopToBottom(treeRoot1)
        print(newArray)

相關連結：

順序列印二叉樹節點：初級版本（當前頁）