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

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

2020-12-31 00:28明柳梦少 Python

这篇文章主要介绍了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