题目
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路
二叉树的层序遍历。
难点在于要求返回一个数组,递归实现起来有困难。
所以实际上是要求非递归实现二叉树的层序遍历。
一个比较好的思路是用队列。初始时若根节点非空,就将其入队,每次出队一个节点,将该节点的值写入数组中,检查其左右孩子,若非空就入队,直至队列为空,二叉树遍历完毕。
代码
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# 返回从上到下每个节点值列表,例:[1,2,3]
def PrintFromTopToBottom(self, root):
# write code here
values = []
if not root:
return values
queue = [root]
while queue:
node = queue.pop(0)
values.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
return values