剑指 Offer 32 - III. 从上到下打印二叉树 III-Python题解

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

例如:
给定二叉树: [3,9,20,null,null,15,7],

返回其层次遍历结果：

[
[3],
[20,9],
[15,7]
]

提示：

节点总数 <= 1000

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        res,node_queue,index_queue = [],collections.deque(),collections.deque()

        # 根节点索引为0
        index_queue.append(0)
        node_queue.append(root)
        self.bfs(index_queue, node_queue, res)
        return res
    def bfs(self, index_queue, node_queue, res):
        if index_queue or node_queue:
            index = index_queue.popleft()
            node = node_queue.popleft()
            if node is not None:
                # 如果索引小于res的长度，说明res已经存储对应层数的数据
                # 层数是偶数的时候是从左到右打印，则往偶数层增加成员时候，都是在尾部插入元素
                # 奇数的时候从右往左打印，则往奇数层增加成员时候，都是在头部插数据
                if index < len(res):
                    item = res[index]
                    if index % 2 == 0:
                        item.append(node.val)
                    else:
                        item.insert(0, node.val)
                else:
                    item = []
                    item.append(node.val)
                    res.append(list(item))
                # 层数的索引加1
                index += 1
                # 二叉树，一次会添加两个成员
                index_queue.append(index)
                node_queue.append(node.left)
                index_queue.append(index)
                node_queue.append(node.right)

            self.bfs(index_queue, node_queue, res)
        else:
            return []