【LeetCode-103】Binary Tree Zigzag Level Order Traversal

这道题只需在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


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值