剑指offer之从上到下打印二叉树（Python）

最新推荐文章于 2024-07-14 14:13:59 发布

明柳梦少

最新推荐文章于 2024-07-14 14:13:59 发布

阅读量3.1k

点赞数

分类专栏：剑指offer（Python语言版）文章标签：算法工程师面试高频题 python

本文链接：https://blog.csdn.net/u010636181/article/details/78289526

版权

剑指offer（Python语言版）专栏收录该内容

15 篇文章 2 订阅

订阅专栏

本文介绍如何解决算法工程师面试中常见的问题——从上到下打印二叉树，采用Python语言进行实现。

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

题目描述

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

思路：用两个list，一个用来装节点，一个用来装节点的value。对于装节点的这个list，我们每次弹出最前面的节点，并依次将左节点和右节点加在后面。不管怎么样，这个装节点的list都能保证，将节点从上到下，同层节点从左到右进行添加、弹出。

代码如下：

class TreeNode():
    def __init__(self,x):
        self.val = x
        self.left = None
        self.right = None
def function(root):
    A = []
    result = []
    if not root:
        return result
    A.append(root)
    while A:
        current_root = A.pop(0)
        result.append(current_root.val)
        if current_root.left:
            A.append(current_root.left)
        if current_root.right:
            A.append(current_root.right)
    return result