二叉树的锯齿形层次遍历
给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
思路:
- 常见思路:根据层次遍历的想法,如果只需要锯齿形,那么其实只要周期性的使一些结果翻转就可以了。
def zigzagLevelOrder(root):
if not root:
return []
queue = [root]
result = []
ifReverse = False
while queue:
size = len(queue)
tempValue = [] # 只需要周期的翻转这个列表就可以了
for i in range(size):
node = queue.pop(0)
tempValue.append(node.val)
if node.left:
queue.apppend(node.left)
if node.right:
queue.append(node.right)
# 得到这一层的结果后,根据条件翻转
if ifReverse:
tempValue.reverse()
ifReverse = False if ifReverse else True
result.append(tempValue)
return result
- 不常见思路:如果我们用栈如何实现呢?分析一下,如果二叉树是[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15],那么第一层是[1];第二层是[3,2],在栈中它是[2,3],为上一层的左右子结点;第三层是[4,5,6,7],在栈中它是[7,6,5,4],为上一层的右左子结点。那么其实就是循环的左右子结点推进入栈,右左子结点推进入栈。但这样还有一个问题,栈是先进先出的,在一个循环里,子结点会影响到父节点,那么为了解决这个问题,我们就需要一个临时保存子结点的栈
def zigzagLevelOrder(root):
if not root:
return []
result = []
stack = [root]
rightLeft = False
stack2 = [] # 临时保存子结点的栈
while stack:
size = len(stack)
temp = []
for i in range(size):
node = stack.pop()
temp.append(node.val)
if rightLeft:
if node.right:
stack2.append(node.right)
if node.left:
stack2.append(node.left)
else:
if node.left:
stack2.append(node.left)
if node.right:
stack2.append(node.right)
rightLeft = False if rightLeft else True
stack = stack2[:]
stack2 = []
result.append(temp)
return result
两种方法的速度和内存使用量也差不多