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

时间:2022-09-20 18:49:02

题目描述

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

思路:

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

2、把每层节点存好:

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

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

代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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

总结

以上就是本文关于Python编程把二叉树打印成多行代码的全部内容,希望对大家有所帮助。感兴趣的朋友可以继续参阅本站其他相关专题,如有不足之处,欢迎留言指出。感谢朋友们对本站的支持!

原文链接:http://blog.csdn.net/u010636181/article/details/78448127