【剑指offer】从上到下打印二叉树（python）

题目描述：
从上往下打印出二叉树的每个节点，同层节点从左至右打印。
解题思路：
二叉树的层次遍历在很多题目中都会出现，解题思路是用一个数组存储每一层的节点，然后弹出每一层的单个节点的同时，插入该节点不为空的左右节点。
完整代码：

# -*- 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 []
        result = [] # 用于存储最终结果
        stack = [root]  # 保存每一层的节点
        while stack:
            length = len(stack)
            for i in range(length):  # 遍历当前层的节点，并从左到右添加左右节点
                temp = stack[0]
                result.append(temp.val)
                if temp.left:
                    stack.append(temp.left)
                if temp.right:
                    stack.append(temp.right)
                stack.remove(temp)
        return result