劍指offer：從上到下打印二叉樹

題目描述

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

結果：[8,6,10,5,7,9,11]

解題思路

典型的使用隊列的題目。每從隊列頭部獲取一個節點，就將該節點的左右子節點存入隊列的尾部。如此往復，直至隊列為空。?

代碼

            
              #coding:utf-8
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)
            
          

leetcode---102.?Binary Tree Level Order Traversal

            
              #coding:utf-8

class Solution:
   #從上往下打印出二叉樹的每個節點，同層節點從左至右打印
    def levelOrder(self, root):
        array = []
        result = []
        if root == None:
            return result
        array.append(root)
        while array:
            templist = []
            for i in range(len(array)):
                newNode = array.pop(0)
                templist.append(newNode.val)
                if newNode.left != None:
                    array.append(newNode.left)
                if newNode.right != None:
                    array.append(newNode.right)
            result.append(templist)
        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.levelOrder(treeRoot1)
        print(newArray)
            
          

leetcode和劍指offer不同之處是輸出的格式，也很好解決，細心一下就好。

[[1], [2, 3], [4, 5, 6], [7, 8]]