剑指offer之把二叉树打印成多行（Python）

题目描述

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

思路：

1、把每层节点的val值用list存好

2、把每层节点存好：

①计算当层节点的个数，这样就保证下一步每层的结点都被pop光

②然后 依次弹出从左到右的每个节点，然后在list中加入该节点对应的左结点、右节点（如果存在的话）

代码如下：

class TreeNode():
    def __init__(self,x):
        self.val = x
        self.left = None
        self.right = None
def function(root):
    result = []
    if not root:
        return result
    A = []
    A.append(root)
    while A:
        temp = []
        size = len(A)
        for Node in A:
            temp.append(Node.val)
        result.append(temp)
        for i in range(size):
            node = A.pop(0)
            if node.left:
                A.append(node.left)
            if node.right:
                A.append(node.right)
    return result