剑指Offer_编程题59:按之字顺序打印二叉树

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

牛客网:链接

有两种方法。第一种:按序获取每一层节点的值;然后将偶数层节点的值倒序。无论哪种方法,都是在剑指Offer_编程题60:把二叉树打印成多行上修改。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回二维列表[[1,2],[4,5]]
    def Print(self, pRoot):
        # write code here
        if not pRoot:
            return []
        curStack = [pRoot]
        result = []
        count = 0
        while curStack:
            '''打印当前行'''
            res = []
            '''存储下一行'''
            nextStack = []
            for i in curStack:
                res.append(i.val)
                if i.left:
                    nextStack.append(i.left)
                if i.right:
                    nextStack.append(i.right)
            '''将存储的下一行直接赋值给新的要遍历的内容'''
            curStack = nextStack
            '''判断是否是奇数'''
            if count & 0x1:
                res = res[::-1]
            result.append(res)
            count += 1
        return result

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

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:  
    def Print(self, pRoot):  
        if not pRoot:  
            return []  
        nodeStack = [pRoot]  
        result = []  
        times = 0
        while nodeStack:
            times += 1
            res = []  
            nextStack = []  
            for i in nodeStack:  
                if times % 2 == 0:
                    res.insert(0, i.val)
                else:
                    res.append(i.val)  
                if i.left:  
                    nextStack.append(i.left)  
                if i.right:  
                    nextStack.append(i.right)  
            nodeStack = nextStack                        
            result.append(res)  
        return result  

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值