“之”字形输出二叉树即第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