“之”字形输出二叉树 python

  “之”字形输出二叉树即第1层从左到右输出,第2层从右到左输出,第3层从左到右输出......第2*n层从右到左输出,第2*n+1层从左到右输出,如下图所示二叉树,“之”字形输出的结果为:1-3-2-4-5-6-7。如果将从左到右输出定义为正向输出,那么奇数层为正向输出,偶数层为反向输出,因此可以通过判断输出输出层数的奇偶性判断输出方向,但实际上奇数层与偶数层交替出现,因此只需设置一个标志变量,每遍历一层,就将标志变量取反即可。程序中除了一个列表遍历到的节点的值,还需要两个临时列表,一个用于存放最新遍历的层次节点,然后对这些节点的子节点继续遍历存入另一个列表,按照标志变量的值,将该列表正向或反向存入结果。

具体代码

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
class Solution:
    #之字形输出二叉树
    def Print(self, pRoot):
        # write code here
        if not pRoot:
            return None
        q=[[pRoot.val]]#最后结果
        reverse=False#输出方向的标志变量
        tmp2=[]
        tmp1=[pRoot.left,pRoot.right]#父节点层
        tmp1=list(filter(lambda c: c is not None,tmp1))#过滤掉空节点
        if tmp1:
            q.append([node.val for node in tmp1[::-1]])
        while 1:
            #先按照正向遍历下一层,然后根据reverse的值,将遍历结果存入q
            while tmp1:
                node=tmp1.pop(0)
                if node.left:
                    tmp2.append(node.left)
                if node.right:
                    tmp2.append(node.right)
            if not tmp2:
                break
            if reverse:
                q.append([node.val for node in tmp2[::-1]])
            else:
                q.append([node.val for node in tmp2])
            reverse= not reverse#下一层的输出方向改变
            tmp1=tmp2
            tmp2=[]
        return q

 

 

posted @ 2019-04-09 21:49  Alice_鹿_Bambi  阅读(1128)  评论(0编辑  收藏  举报