103. 二叉树的锯齿形层序遍历

  1. 题目链接

  2. 解题思路:和层序遍历有明显的不同。通过观察,可以得到,当前层,和下一层的顺序是「相反」的,遇到这种相反的问题,考虑用栈。本题就是用两个栈,一个栈在放入时,先放左儿子,再放右儿子,另一个栈在放入时,先放右儿子,再放左儿子。然后两个栈交替使用即可。

    • 为什么能得到这个思路?观察得到的
  3. 代码

    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
    
posted @ 2024-12-25 17:00  ouyangxx  阅读(21)  评论(0)    收藏  举报