请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
层序遍历:
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def Print(self, pRoot):
# write code here
res = []
if not pRoot:
return res
stack, flag = [pRoot], 1
while stack:
ceng = []
for _ in range(len(stack)):
node = stack.pop(0)
ceng.append(node.val)
if node.left:
stack.append(node.left)
if node.right:
stack.append(node.right)
if flag == -1:
ceng.reverse()
flag *= -1
res.append(ceng)
return res