题目: 二叉树的锯齿形层次遍历
给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层次遍历如下:
[
[3],
[20,9],
[15,7]
]
来源:力扣(LeetCode)第103题
链接:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
分析:
层次遍历比较简单,主要是锯齿形,就是单数层从左到右,双数层从右到左。网上有很多题解,看了他们的之后觉得没有自己的好,他们基本都是使用了反转列表的操作,理论上会很耗时间。所以,我讲的是自己写的思路。
思路:
- 运用双栈法,分别储存单数层和双数层两种情况。
- ans为返回的总列表,res为每一层的答案,stack和helper是两个栈,方法和之前的前、中序遍历差不多。
- 写一个主循环while,再写两个循环放在主循环中,一个遍历单数层,一个遍历双数层。
- 两个循环里把双栈中的节点拿出来,同时也把节点的两个子节点(stack或者helper)以及他们的值(res)也存起来。
- 最后将res的值加到ans中。主循环结束,ans值也都进去了。
代码:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def zigzagLevelOrder(self, root: TreeNode) -> List[List[int]]:
if root is None:
return []
stack = []
helper = [root]
res = []
ans = [[root.val]]
cnt = 1
while stack or helper:
for i in range(cnt):
tail = helper.pop()
if tail.right is not None:
stack.append(tail.right)
res.append(tail.right.val)
if tail.left is not None:
stack.append(tail.left)
res.append(tail.left.val)
if res:
ans.append(res)
res = []
cnt = len(stack)
else:
break
for i in range(cnt):
tail = stack.pop()
if tail.left is not None:
helper.append(tail.left)
res.append(tail.left.val)
if tail.right is not None:
helper.append(tail.right)
res.append(tail.right.val)
if res:
ans.append(res)
res = []
cnt = len(helper)
return ans
复杂度分析:
- 虽然有一个while,两个for,但是其实只遍历了一遍节点,而且没有额外的其他耗时操作,也没有递归
- 所以时间复杂度为O(n) n为树的总节点
- 空间复杂度为O(n)
总结:
双栈法的另一个用处,当遇到在一个循环里但是其中两次循环的条件不一眼时,可以写两个循环分别运算,再用一个主循环让两个循环不会结束。