天天看点

《剑指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