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

感谢你能够认真阅读完这篇文章，希望小编分享的“Python中如何把二叉树打印成多行代码”这篇文章对大家有帮助，同时也希望大家多多支持亿速云，关注亿速云行业资讯频道，更多相关知识等着你来学习!