103. 二叉树的锯齿形层序遍历
-
解题思路:和层序遍历有明显的不同。通过观察,可以得到,当前层,和下一层的顺序是「相反」的,遇到这种相反的问题,考虑用栈。本题就是用两个栈,一个栈在放入时,先放左儿子,再放右儿子,另一个栈在放入时,先放右儿子,再放左儿子。然后两个栈交替使用即可。
- 为什么能得到这个思路?观察得到的
-
代码
class Solution: def zigzagLevelOrder(self, root: Optional[TreeNode]) -> List[List[int]]: if root == None: return [] stack1 = [] # 使用stack1时,先放右儿子,后放左儿子 stack2 = [] # 使用stack2时,先放左儿子,再放右儿子 isOne = True # True表示用stack1 False 表示用stack2 ans = [] # 最终的结果 tmp = [] # 每层的结果 stack1.append(root) while stack1 or stack2: if isOne: isOne = False tmp.clear() while stack1: cur = stack1.pop() tmp.append(cur.val) if cur.left: stack2.append(cur.left) if cur.right: stack2.append(cur.right) ans.append(copy.copy(tmp)) else : isOne = True tmp.clear() while stack2: cur = stack2.pop() tmp.append(cur.val) if cur.right: stack1.append(cur.right) if cur.left: stack1.append(cur.left) ans.append(copy.copy(tmp)) return ans

浙公网安备 33010602011771号