LeetCode Z字形遍历二叉树

利用队列，从前向后在队列中取数据时，此时向队列中插入数据，插在队列尾部；从后向前在队列中取数据时，此时若插入数据，插在队列头部。感觉这种思路比较容易想到。时间还能接受

from collections import deque
class Solution:
    def zigzagLevelOrder(self, root: TreeNode) -> List[List[int]]:
        res = []
        if not root:
            return res
        q = deque([root, ])
        count = 0          #记录层数
        status = True    #用于判断从队列中读取数据的顺序
        while q:
            res.append([])
            if status:
                for _ in range(len(q)):  
                    node = q.popleft()   #从前向后在队列中取数据，则往队列中插入时，添加在队列尾部
                    res[count].append(node.val)
                    if node.left:
                        q.append(node.left)
                    if node.right:
                        q.append(node.right)
            else:
                for _ in range(len(q)):
                    node = q.pop()    #从后向前在队列中取数据，则往队列中插入数据时，添加在队列头部
                    res[count].append(node.val)
                    if node.right:
                        q.insert(0, node.right)
                    if node.left:
                        q.insert(0, node.left)
            count += 1 
            status = not status
        return res