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

题目描述

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

思路：

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

2、把每层节点存好：

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

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

代码如下：

class TreeNode():    def __init__(self,x):        self.val = x        self.left = None        self.right = Nonedef 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