题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
思路:
两个栈,分二级,第一级存一行的数curStack,第二级存着一行数所对应的left,right的值nextStack,每次结束
一轮循环,把curStack的值给result,然后再将nodes设成nextStack继续进行循环。
比上道题按照之字形打印二叉树更简单,连结合层数都省了。
参考答案:
# -*- 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 = []
current_layer = [pRoot]
while current_layer:
layer_res = []
next_layer = []
for node in current_layer:
layer_res.append(node.val)
if node.left:
next_layer.append(node.left)
if node.right:
next_layer.append(node.right)
res.append(layer_res[:])
current_layer = next_layer[:]
return res
Note
- 注意,其中current_layer 的结果
current_layer = next_layer[:]
。