leetcode之Binary Tree Zigzag Level Order Traversal

howtogetout

于 2015-12-13 12:43:33 发布

阅读量227

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/howtogetout/article/details/50282175

本文介绍了一个简单的二叉树锯齿形层次遍历算法实现，通过加入奇偶数判断开关来控制每层节点值的顺序，实现正序与逆序交替输出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这题就是在之前的基础上加个奇偶数的判断开关。不难。代码如下：

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

class Solution(object):
    def zigzagLevelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if  not root:
            return []
        a = []
        queue = [root]
        b = 0
        while queue:
            if b == 0:
                a.append([q.val for q in queue])
            else:
                a.append([q.val for q in queue][::-1])
            queue = [q for node in queue for q in (node.left, node.right) if q]
            if b == 0:
                b = 1
            else:
                b = 0
        return a