题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路:
与逐层打印二叉树一样,只是多了层数判定,依据层数做判定输出。
按之字形顺序打印二叉树需要两个栈。我们在打印某一行节点时,把下一层的子节点保存到相应的栈里。保存层数,对temp_res
的结果保存时结合层数做正反序append
。
参考答案:
# -*- 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
if not pRoot:
return []
current_layer = [pRoot]
res = []
layer_level = 0
while current_layer:
next_layer = []
layer_res = []
for node in current_layer:
layer_res.append(node.val)
if node.left:
next_layer.append(node.left)
if node.right:
next_layer.append(node.right)
current_layer = next_layer[:]
if layer_level % 2 == 1:
res.append(layer_res[::-1])
else:
res.append(layer_res[:])
layer_level += 1
return res
Note
- 这道题要十分注意,其中current_layer 的结果
current_layer = next_layer[:]
,即是完全复制,直接用current_layer = next_layer
会出现错误的。