题解 | #把二叉树打印成多行#
把二叉树打印成多行
http://www.nowcoder.com/practice/445c44d982d04483b04a54f298796288
广度优先搜索(层次遍历),借助队列。
与#从上往下打印二叉树#类似,多了一个len(queue),记录每层节点的个数。 https://blog.nowcoder.net/n/ae23b78c29f449e5a4fcb1ad94292a63
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
#
# @param pRoot TreeNode类
# @return int整型二维数组
#
class Solution:
def Print(self , pRoot: TreeNode) -> List[List[int]]:
if pRoot==None:
return []
queue=[]
result=[]
queue.insert(0,pRoot)
while len(queue)>0:
row=[]
n=len(queue)
for i in range(n):#目前队列的长度,就是每层节点的个数
node=queue.pop()#出队,访问
row.append(node.val)
if node.left:
queue.insert(0,node.left)
if node.right:
queue.insert(0,node.right)
result.append(row)
return result