从上往下打印二叉树

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

思路:这是一个二叉树的层次遍历,这时我们需要用一个队列来实现

注意:这里我们需要注意的是root为None的情况

# -*- 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
        queue = []
        arr = []
        if root is None:
            return arr
        queue.append(root)
        while len(queue)!=0:
            temp = queue[0]
            del queue[0]
            if temp.left!=None:
                queue.append(temp.left)
            if temp.right!=None:
                queue.append(temp.right)
            arr.append(temp.val)
        return arr