二叉树的锯齿形层次遍历

二叉树的锯齿形层次遍历

给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

思路:

  1. 常见思路:根据层次遍历的想法,如果只需要锯齿形,那么其实只要周期性的使一些结果翻转就可以了。
def zigzagLevelOrder(root):
    if not root:
        return []
    queue = [root]
    result = []
    ifReverse = False
    while queue:
        size = len(queue)
        tempValue = [] # 只需要周期的翻转这个列表就可以了
        for i in range(size):
            node = queue.pop(0)
            tempValue.append(node.val)
            if node.left:
                queue.apppend(node.left)
            if node.right:
                queue.append(node.right)
        # 得到这一层的结果后,根据条件翻转
        if ifReverse:
            tempValue.reverse()
        ifReverse = False if ifReverse else True
        result.append(tempValue)
    return result
  1. 不常见思路:如果我们用栈如何实现呢?分析一下,如果二叉树是[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15],那么第一层是[1];第二层是[3,2],在栈中它是[2,3],为上一层的左右子结点;第三层是[4,5,6,7],在栈中它是[7,6,5,4],为上一层的右左子结点。那么其实就是循环的左右子结点推进入栈,右左子结点推进入栈。但这样还有一个问题,栈是先进先出的,在一个循环里,子结点会影响到父节点,那么为了解决这个问题,我们就需要一个临时保存子结点的栈
def zigzagLevelOrder(root):
    if not root:
        return []
    result = []
    stack = [root]
    rightLeft = False
    stack2 = [] # 临时保存子结点的栈
    while stack:
        size = len(stack)
        temp = []
        for  i in range(size):
            node = stack.pop()
            temp.append(node.val)
            if rightLeft:
                if node.right:
                    stack2.append(node.right)
                if node.left:
                    stack2.append(node.left)
            else:
                if node.left:
                    stack2.append(node.left)
                if node.right:
                    stack2.append(node.right)
        rightLeft = False  if rightLeft else True
        stack = stack2[:]
        stack2 = []
        result.append(temp)
    return result

两种方法的速度和内存使用量也差不多

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值