剑指offer 按之字形顺序打印二叉树（双队列层序遍历）

题目描述

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

Solution

在常规层序遍历改进而来。使用一个栈保存下一次要访问的各个节点，只是入栈时变换左右节点。

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root: return []
        res, queue, = [], collections.deque()
        queue.append(root)
        while queue:
            tmp, queue_next = collections.deque(), collections.deque()
            for node in queue:
                if (len(res)%2) == 0: tmp.append(node.val)
                else: tmp.appendleft(node.val)
                
                if node.left: queue_next.append(node.left)
                if node.right: queue_next.append(node.right)
            queue = queue_next
            res.append(list(tmp))
        return res