从上往下打印出二叉树的每个节点,同层节点从左至右打印。
基本思路:借助队列。
每一次打印一个结点的时候,如果该结点有子结点,则把该结点的子结点加入到一个队列的末尾。接下来到队列的头部取出最早进入队列的结点,重复前面的操作。
# -*- 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
array = []
root_list = []
if root:
i = 0
root_list.append(root)
while i < len(root_list):
array.append(root_list[i].val)
if root_list[i].left:
root_list.append(root_list[i].left)
if root_list[i].right:
root_list.append(root_list[i].right)
i += 1
return array