[Leetcode]Binary Tree Zigzag Level Order Traversal

4 篇文章 0 订阅

Given a binary tree, return the zigzag level order traversal of its nodes' values. (ie, from left to right, then right to left for the next level and alternate between).

For example:
Given binary tree {3,9,20,#,#,15,7},

    3
   / \
  9  20
    /  \
   15   7

return its zigzag level order traversal as:

[
  [3],
  [20,9],
  [15,7]
]
和Binary Tree Level Order Traversal类似,也是树的层序遍历,只不过是在偶数层自左向右遍历,奇数层自右向左遍历~可以直接在 Binary Tree Level Order Traversal的代码上做改进,把奇数层的结果数列reverse一下~或者用下面这种解法也是可以的~同时维护两个栈,一个栈用来读取,一个栈用来存放下一层的节点

class Solution:
    # @param root, a tree node
    # @return a list of lists of integers
    def zigzagLevelOrder(self, root):
        if root is None: return []
        cur, res, vals = [root], [], [root.val]
        while cur:
            res.append(vals)
            next, vals = [], []
            while cur:
                node = cur.pop()
                if len(res) % 2 == 0:
                    if node.left:
                        vals.append(node.left.val)
                        next.append(node.left)
                    if node.right:
                        vals.append(node.right.val)
                        next.append(node.right)
                else:
                    if node.right:
                        vals.append(node.right.val)
                        next.append(node.right)
                    if node.left:
                        vals.append(node.left.val)
                        next.append(node.left)
            cur = next
        return res



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值