《剑指offer》之从上往下打印二叉树（python实现）

 

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

思路分析：
从题意看这就是二叉树的层次遍历，宽度优先搜索问题。
借助两个数组，一个用来存储节点（这是一个队列），另外一个用来存储节点值（result）。
从根节点开始遍历节点，把每一层的节点都放进这个队列中；
开始遍历这个队列，将队列的值以及放进result数组中最后返回。
总结一下：
1）将根节点加入队列
2）队列不为空时取队首元素的值append到result数组
3）将下一层节点加入队列队尾
4）返回第二步，直到队列为空结束

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 []
        queue = [root]
        result = []
        while len(queue) > 0:
            Hqueue = queue.pop(0)
            result.append(Hqueue.val)
            if Hqueue.left:
                queue.append(Hqueue.left)
            if Hqueue.right:
                queue.append(Hqueue.right)
        return result