把二叉树打印成多行

题目

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

思路

• 解法一：队列+层次遍历
  利用队列的先进先出特性，可以模拟二叉树的层次遍历。
  将根节点入队，打印结点值，将其左节点和右节点先后入队。
  将队列中的元素从index=0处，出队，打印结点的值，将各个结点的左右结点先后入队。
  以此往复，实现二叉树的层次遍历。

代码

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回二维列表[[1,2],[4,5]]
    def Print(self, pRoot):
        # write code here
        if not pRoot:
            return []
        res = []
        stack = [pRoot]
        while len(stack) != 0:
            node_list = []
            while len(stack)!= 0:
                node_list.append(stack.pop(0))
            res.append([x.val for x in node_list])
            for each_node in node_list:
                if each_node.left is not None:
                    stack.append(each_node.left)
                if each_node.right is not None:
                    stack.append(each_node.right)
        return res