【LeetCode-103】Binary Tree Zigzag Level Order Traversal

简单调整后解决复杂问题：树的螺旋遍历

最新推荐文章于 2020-07-25 07:42:38 发布

原创最新推荐文章于 2020-07-25 07:42:38 发布 · 250 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode-103 #Binary Tree Zigzag L #python

leetcode 专栏收录该内容

80 篇文章

订阅专栏

本文介绍了一种简化后的树的螺旋遍历算法，通过在基本框架上进行细微调整，实现对二叉树进行螺旋顺序遍历。详细解释了算法的核心思想、实现步骤以及代码实现，并通过实例演示了其应用。

这道题只需在102的基础上稍微变化下就行，很简单了，fighting

class Solution(object):
    def zigzagLevelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        # 最终结果
        res = []
        # 存放每一层结果
        levelRes = []

        if root == None:
            return res

        # 当前一层的节点
        current = []
        # 下一层的节点
        next = []

        # 根节点加入
        current.append(root)

        # 标记是正向还是反向来
        flag = True

        while len(current) > 0:
            node = current.pop()
            
            if flag:
                # （正向，则先添加左节点，再添加右节点）
                if node.left != None:
                    next.append(node.left)

                if node.right != None:
                    next.append(node.right)
            else :
                # (反向。先添加右节点，再添加左节点)
                if node.right != None:
                    next.append(node.right)

                if node.left != None:
                    next.append(node.left)


            # 加入访问节点的值
            levelRes.append(node.val)

            if len(current) == 0:
                current = next
                next = []
                res.append(levelRes)
                levelRes = []

                # 将flag取反
                flag = not flag

        return res