网络编程
位置:首页>> 网络编程>> Python编程>> Python编程把二叉树打印成多行代码

Python编程把二叉树打印成多行代码

作者:明柳梦少  发布时间:2023-06-24 15:12:57 

标签:python,二叉树,遍历

题目描述

从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。

思路:

1、把每层节点的val值用list存好

2、把每层节点存好:

①计算当层节点的个数,这样就保证下一步每层的结点都被pop光

②然后依次弹出从左到右的每个节点,然后在list中加入该节点对应的左结点、右节点(如果存在的话)

代码如下:


class TreeNode():
 def __init__(self,x):
   self.val = x
   self.left = None
   self.right = None
def function(root):
 result = []
 if not root:
   return result
 A = []
 A.append(root)
 while A:
   temp = []
   size = len(A)
   for Node in A:
     temp.append(Node.val)
   result.append(temp)
   for i in range(size):
     node = A.pop(0)
     if node.left:
       A.append(node.left)
     if node.right:
       A.append(node.right)
 return result

来源:http://blog.csdn.net/u010636181/article/details/78448127

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com