剑指offer面试题【32】----从上到下打印二叉树【队列】【举例让抽象问题具体化】【Python】

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

思路

从上到下打印二叉树的规律：

每次打印一个节点后，如果该节点有子结点，则把该节点的子节点放到一个队列的末尾。接下来到队列的头部取出最早进入队列的节点，重复前面的打印操作，知直到队列的所有节点都被打印出来。

代码实现

class TreeNode:
     def __init__(self, x): 
         self.val = x
         self.left = None
         self.right = None                
class Solution:
    # 返回从上到下每个节点值列表，例：[1,2,3]
    def PrintFromTopToBottom(self, root): 
        l=[]
        if root==None:
            return None
        q=[root]
        while len(q):
            t=q.pop(0)
            l.append(t.val)
            if t.left:
                q.append(t.left)
            if t.right:
                q.append(t.right)
        return l
#测试
node1=TreeNode(8)    
node2=TreeNode(6)
node3=TreeNode(10)
node4=TreeNode(5)
node5=TreeNode(7)
node6=TreeNode(9)
node7=TreeNode(11)

node1.left=node2
node1.right=node3
node2.left=node4
node2.right=node5
node3.left=node6
node3.right=node7

S=Solution()
temp=S.PrintFromTopToBottom(node1)
print(temp)