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