刷题--从上往下打印二叉树

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

基本思路:借助队列。
每一次打印一个结点的时候,如果该结点有子结点,则把该结点的子结点加入到一个队列的末尾。接下来到队列的头部取出最早进入队列的结点,重复前面的操作。

# -*- 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
        array = []
        root_list = []

        if root:
            i = 0
            root_list.append(root)

            while i < len(root_list):
                array.append(root_list[i].val)

                if root_list[i].left:
                    root_list.append(root_list[i].left)
                if root_list[i].right:
                    root_list.append(root_list[i].right)

                i += 1

        return array

猜你喜欢

转载自blog.csdn.net/treasure_z/article/details/80217793