【牛客剑指offer刷题】：Python：22.从上往下打印二叉树

从上往下打印二叉树

时间限制：1秒 空间限制：32768K 热度指数：324215
算法知识视频讲解

题目描述

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

代码

实际就是广度优先搜索 BFS, 借助一个队列就可以实现。

# -*- 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
        res = [root]
        if not root:
            return []
        ans = []
        while res:
            line = res.pop(0) 
            ans.append(line.val)
            if line.left:
                res.append(line.left)
            if line.right:
                res.append(line.right)
        return ans