剑指offer 之从上到下打印二叉树

"""
问题：从上到下打印二叉树
    从上往下打印出二叉树的每个节点，同层节点从左至右打印。
思路：
    1.队列思想,先进先出, append() + pop(0)
    2.从根节点开始循环判断左右结点
"""


# -*- 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 []
        mid_list = [root]
        res = []
        # 此处注意应该判断队列中是否还有元素
        # 我第一次写的是 root 引以为戒【捂脸】
        while mid_list:
            node = mid_list.pop(0)
            res.append(node.val)
            if node.left:
                mid_list.append(node.left)
            if node.right:
                mid_list.append(node.right)
        return res