老卫带你学---剑指offer刷题系列(60.把二叉树打印成多行)

60.把二叉树打印成多行

问题：

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

解决：

思想：

层次遍历的变种，在每层中间加入一个row中间数组，用来保存该层的数据，另外将下一层的节点加入队列中。

python代码：

# -*- 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 []
        result=[]
        tmp=[pRoot]
        while tmp:
            n=len(tmp)
            row=[]
            for i in tmp:
                row.append(i.val)
            result.append(row)
            
            for j in range(n):
                u=tmp.pop(0)
                if(u.left):
                    tmp.append(u.left)
                if(u.right):
                    tmp.append(u.right)
        return result