剑指offer:按之字形顺序打印二叉树-Python

转载自:https://blog.csdn.net/u010005281/article/details/79759926

题目描述

请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。

解题思路

先给定一个二叉树的样式: 
这里写图片描述 
前段时间做过一道题,其要求是: 
从上往下打印出二叉树的每个节点,同层节点从左至右打印。要求的输出样式是:[1, 2, 3, 4, 5, 6, 7]。这样一道题只需要按序将节点打印出来即可,可利用列表,将新节点存入列表的尾端,并从列表的头部依次打印存入的节点值即可,具体代码见链接。 
而这道题所不同的地方是:其要求输出的样式是:[[1], [3,2], [4,5,6,7]]。和上段中的题目相比,这道题不仅要求按序输出节点值,还要求包含以下信息: 
1. 每一层所包含的树节点; 
2. 偶数层的树节点需倒序。 
所以求解这道题的思路可以说是上一题的升级版,我们同样可以利用列表存储节点,但不同的是列表每次只存储一层的节点,在遍历当前层节点的同时,存储下一层的节点,以此类推。面对要求的偶数层倒序,亦有两种解题思路,第一种是:获取到所有节点的值后,将偶数层的节点值倒序。第二种则是在获取节点的值的时候就倒序存入。以下给出两种方式的Python代码:

代码

方式1: 
1. 按序获取每一层节点的值; 
2. 将偶数层节点的值倒序。

def Print(self, pRoot):
    resultArray = []
    if not pRoot:
        return resultArray
    curLayerNodes = [pRoot]
    isEvenLayer = True
    while curLayerNodes:
        curLayerValues = []
        nextLayerNodes = []
        isEvenLayer = not isEvenLayer
        for node in curLayerNodes:
            curLayerValues.append(node.val)
            if node.left:
                nextLayerNodes.append(node.left)
            if node.right:
                nextLayerNodes.append(node.right)
        curLayerNodes = nextLayerNodes
        resultArray.append(curLayerValues[::-1]) if isEvenLayer else resultArray.append(curLayerValues)
    return resultArray

方式2: 
获取每一层的节点的值时,如果是偶数层,则将每个节点的值插入到列表的头部,即实现了获取节点值时如果是偶数层则倒序排列的效果:

def Print2(self, pRoot):
    resultArray = []
    if not pRoot:
        return resultArray
    curLayerNodes = [pRoot]
    isEvenLayer = True
    while curLayerNodes:
        curLayerValues = []
        nextLayerNodes = []
        isEvenLayer = not isEvenLayer
        for node in curLayerNodes:
            curLayerValues.insert(0, node.val) if isEvenLayer else curLayerValues.append(node.val)
            if node.left:
                nextLayerNodes.append(node.left)
            if node.right:
                nextLayerNodes.append(node.right)
        curLayerNodes = nextLayerNodes
        resultArray.append(curLayerValues)
    return resultArray

相关链接:

顺序打印二叉树节点:初级版本

顺序打印二叉树节点:中级版本

顺序打印二叉树节点:高级版本(当前页)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值