题目
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
思路
- 解法一:队列+层次遍历
利用队列的先进先出特性,可以模拟二叉树的层次遍历。
将根节点入队,打印结点值,将其左节点和右节点先后入队。
将队列中的元素从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