一旦看到这种有关顺序的,第一个就要想到用栈或队列实现,有这个思路才能进一步实现、优化。
比如这道题,要求 “第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印”,显然,如果能直接利用栈或队列的特性,就能实现题目要求的各种顺序了。
算法流程:
- 维护两个栈,一个存放奇数层的节点,一个存放偶数层的节点
- 分奇数层和偶数层处理
- 奇数层按顺序,每次偶数层的栈保存偶数层节点,先存左结点,这样等下次pop时就是右节点先打印
- 偶数层反序,每次奇数层的栈保存偶数层节点,先存右结点,这样等下次pop时就是左节点先打印
#####需要注意的是,append时候的是 节点TreeNode 还是 TreeNode.val
# -*- 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 pRoot == None:
return []
stack1 = [pRoot]
stack2 = []
res1 = []
res2 = []
res = []
while stack1 or stack2:
#处理奇数层
while stack1:
for i in range(len(stack1)):#写这一行,stack pop {8,6,10,5,7,9,11}的输出就会是[[8],[6],[10],[5],[7],[9],[11]]而不是[[8],[6,10],[5,7,9,11]]
node = stack1.pop()
# 下一层偶数层按顺序,先存左节点,再存右节点,pop的时候才能先右后左
res1.append(node.val)
if node.left:
stack2.append(node.left)
if node.right:
stack2.append(node.right)
# 收集当前层的结果
#if res1 != []:
res.append(res1)
#把res1清空,避免下次迭代时候值重复
res1 = []
while stack2:
for i in range(len(stack2)):
node = stack2.pop()
# 下一层奇数层按顺序,先存右节点,再存左节点,pop的时候才能先左后右
res2.append(node.val)
if node.right:
stack1.append(node.right)
if node.left:
stack1.append(node.left)
#if res2 != []:
res.append(res2)
#把res2清空,避免下次迭代时候值重复
res2 = []
return res
# 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 pRoot == None:
return []
stack1 = [pRoot]
stack2 = []
res1 = []
res2 = []
res = []
while stack1 or stack2:
#处理奇数层
while stack1:
for i in range(len(stack1)):#写这一行,stack pop {8,6,10,5,7,9,11}的输出就会是[[8],[6],[10],[5],[7],[9],[11]]而不是[[8],[6,10],[5,7,9,11]]
node = stack1.pop()
# 下一层偶数层按顺序,先存左节点,再存右节点,pop的时候才能先右后左
res1.append(node.val)
if node.left:
stack2.append(node.left)
if node.right:
stack2.append(node.right)
# 收集当前层的结果
#if res1 != []:
res.append(res1)
#把res1清空,避免下次迭代时候值重复
res1 = []
while stack2:
for i in range(len(stack2)):
node = stack2.pop()
# 下一层奇数层按顺序,先存右节点,再存左节点,pop的时候才能先左后右
res2.append(node.val)
if node.right:
stack1.append(node.right)
if node.left:
stack1.append(node.left)
#if res2 != []:
res.append(res2)
#把res2清空,避免下次迭代时候值重复
res2 = []
return res