二叉树按照宽度优先遍历就是使用队列进行,具体在图解算法中有讲解。
如果要按照行来打印二叉树,需要多加入一个last、和next_last变量。
如上图二叉树。定义last为目前行的最右节点,定义next_last为下一行的最右节点。因此每次打印到last时换行,再更新last和next_last,就可以实现按行打印二叉树。
具体:
1.令last等于节点1,1入队
2.1出队打印,1的子节点2入队,next_last指向2;1的子节点3入队,next_last指向3.(next_last永远指向最新入队的节点)
3.打印的1与last的值相等则换行,last = next_last
4.2出队打印,2的子节点4入队,next_last指向4
5.打印的2与last的值不相等则不换行,继续出队打印下一节点。
6….循环执行以上步骤。
二叉树被记录成文件的过程称为二叉树的序列化。
李如下图先序序列化:
头节点1 2 一个字符结束 加上“!”,然后是左孩子,3,有一个字符结束 加上“!”,然后是3的左孩子,是空,因此是一个“#”,与字符结束符号“!”,然后是3的右孩子,同理是“#!”, 然后是12的右孩子,同理是“#!”。
总结来说,就是在遍历二叉树的时候,每一个字符结束就是用“!”间隔,遇到空节点就是用“#”占位。
作业:
有一棵二叉树,请设计一个算法,按照层次打印这棵二叉树。
给定二叉树的根结点root,请返回打印结果,结果按照每一层一个数组进行储存,所有数组的顺序按照层数从上往下,且每一层的数组内元素按照从左往右排列。保证结点数小于等于500。
我的实现:
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class TreePrinter:
def printTree(self, root):
# write code here
queue = []
result = []
res_line = []
last = root
queue.append(root)
nlast = None
while queue:
now_node = queue.pop(0)
res_line.append(now_node.val)
if now_node.left:
queue.append(now_node.left)
nlast = now_node.left
if now_node.right:
queue.append(now_node.right)
nlast = now_node.right
if now_node == last:
last = nlast
result.append(res_line)
res_line = []
return result