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

最新推荐文章于 2022-05-09 17:44:59 发布

breeze_blows

最新推荐文章于 2022-05-09 17:44:59 发布

阅读量180

点赞数

分类专栏： LeetCode 文章标签：从上到下打印二叉树 III 二叉树 LeetCode 题解

本文链接：https://blog.csdn.net/breeze_blows/article/details/108154684

版权

LeetCode 专栏收录该内容

41 篇文章 1 订阅

订阅专栏

链接：剑指 Offer 32 - III. 从上到下打印二叉树 III

层次遍历二叉树的基础上每次用一个数组保存当前层的结点，然后按照奇偶层进行反序就好了

# 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]]:
        if not root:
            return []
        queue = collections.deque()
        queue.append(root)
        res = []
        cnt = 1
        while queue:
            tmp = []
            size = len(queue)
            for _ in range(size):
                q = queue.popleft()
                tmp.append(q.val)
                if q.left:
                    queue.append(q.left)
                if q.right:
                    queue.append(q.right)
            if cnt == 1:
                res.append(tmp)
            else:
                res.append(tmp[::-1])
            cnt = -cnt
        return res