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

最新推荐文章于 2022-02-23 15:37:42 发布

dataRookieUp

最新推荐文章于 2022-02-23 15:37:42 发布

阅读量150

点赞数

本文链接：https://blog.csdn.net/weixin_43988882/article/details/103326661

版权

本文介绍了如何使用Python解决从上至下打印二叉树的问题，通过利用队列进行层次遍历，按照先进先出原则，依次打印各层节点，先左子节点后右子节点。

摘要由CSDN通过智能技术生成

题目描述

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

知识点：树，队列

解题思路

这题我们可以通过使用一个队列保存我们需要打印的节点，然后按照先进先出的原则提取队列中的节点。
具体思路是：每次打印一个节点时，如果该节点有子节点，则把该节点的子节点放在队列的末尾。接下来从队列的头部取出最早进入队列的节点，重复上面的过程，直至队列中所有节点都被打印出来。

Python 代码

class Solution:
    # 返回从上到下每个节点值列表，例：[1,2,3]
    def PrintFromTopToBottom(self, root):
        # write code here
        if root is None:
            return []
        queue = [root]
        printarr = []
        while queue:
            root = queue.pop(0)
            printarr.append(root.val)
            if root.left :
                queue.append(root.left)
            if root.right:
                queue.append(root.right)
        return printarr