题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
利用队列就可以实现,主要注意的是牛客网的返回列表要求元素只有数值,不是结点
# -*- 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
if not root:
return []
queue=[]
tree=[]
front=0
tail=1
queue.append(root)
while front!=tail:
node = queue.pop(0)
tree.append(node.val)
if node.left!=None:
queue.append(node.left)
tail+=1
if node.right!=None:
queue.append(node.right)
tail+=1
front+=1
return tree