二叉树打印

二叉树按照宽度优先遍历就是使用队列进行,具体在图解算法中有讲解。

如果要按照行来打印二叉树,需要多加入一个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

猜你喜欢

转载自blog.csdn.net/yinruiyang94/article/details/79423923