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


    题目描述

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

    前段时间做过一道题,其要求是:从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。输出是形如[[1], [2,3], [4,5,6,7]]。

    而这道题所不同的地方是:其要求输出的样式是:[[1], [3,2], [4,5,6,7]]。和上段中的题目相比,这道题不仅要求按序输出节点值,还要求包含以下信息:

    1. 每一层所包含的树节点;

    2. 偶数层的树节点需倒序。

    所以求解这道题的思路可以说是上一题的升级版,我们同样可以利用列表存储节点,但不同的是列表每次只存储一层的节点,在遍历当前层节点的同时,存储下一层的节点,以此类推。面对要求的偶数层倒序,亦有两种解题思路,第一种是:获取到所有节点的值后,将偶数层的节点值倒序。第二种则是在获取节点的值的时候就倒序存入。以下给出两种方式的Python代码:

    方式1:

    1. 按序获取每一层节点的值;

    2. 将偶数层节点的值倒序。

    # -*- coding:utf-8 -*-
    # class TreeNode:
    #     def __init__(self, x):
    #         self.val = x
    #         self.left = None
    #         self.right = None
    class Solution:
        def Print(self, pRoot):
            # write code here
            if not pRoot:
                return []
            resultArray = []
            curLayerNodes = [pRoot]
            isEvenLayer = True#Even:偶数
            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:

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

    # -*- coding:utf-8 -*-
    # class TreeNode:
    #     def __init__(self, x):
    #         self.val = x
    #         self.left = None
    #         self.right = None
    class Solution:
        def Print(self, pRoot):
            # write code here
            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
    
  • 相关阅读:
    电影
    Python中__init__方法介绍
    学习笔记(1?)
    python参数传递
    #!/usr/bin/env python与#!/usr/bin/python
    面向对象程序设计
    Python模块学习 random 随机数生成
    正则表达式前缀r
    python 去除 list 重复元素
    python datetime处理时间
  • 原文地址:https://www.cnblogs.com/tianqizhi/p/9697138.html
Copyright © 2020-2023  润新知