面试题32:从上到下打印二叉树

题目一:不分行从上到下打印二叉树
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。

解题思路:
总结规律为首先将根节点加入到队列中,然后当队列不为空时,打印队列首部的的节点值,如果该节点有左右节点则将左右节点加入队列的尾部,接着需要再将队列首部的元素取出,重复上述的打印操作。

代码实现:

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回从上到下每个节点值列表,例:[1,2,3]
    def PrintFromTopToBottom(self, root):
        # write code here
        if root is None:
            return []  #如果为空,直接返回空列表
        #保存节点的缓存队列
        res = []
        res_val = []
        res.append(root)
        while len(res)>0:
            Node = res.pop(0)        #将队列头部的节点取出
            res_val.append(Node.val)#加入打印的列表的尾端
            if Node.left:            #如果该节点有左节点将其加入队列尾端
                res.append(Node.left)
            if Node.right:           #如果该节点有右节点将其加入队列尾端
                res.append(Node.right)
        return res_val              #最终返回值列表

提交结果:

————————————————————————————————
题目二:
分行打印按照层次遍历的结果

解题思路:
这道题和上一题目思路一样,但是多了一个换行控制,将同一层的val添加到一个列表,当需要换行时把这一列表的值全部添加到打印列表,然后进行下一层的列表添加。为此需要引入两个变量,一个是当前层需要打印的节点数,一个是下一层的总节点数,当当前层次需要打印的节点数为0是进行当前层的列表添加操作并且进行变量更新以便于下层的打印。

代码实现:

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回从上到下每个节点值列表,例:[1,2,3]
    def Print(self, pRoot):
        # write code here
        if pRoot ==  None:
            return []
        res = []
        res_val = []
        res.append(pRoot)

        ToBePrint = 1 #根节点为一个节点
        NextLevel = 0 #下一层的节点数
        temp = []     #储存每层节点的变量 
        while len(res)>0:
            
            node = res.pop(0)
            temp.append(node.val)
            ToBePrint -=1    #要打印的数目减1

            if node.left:
                res.append(node.left)
                NextLevel+=1  #x下一层节点数加1
            if node.right:
                res.append(node.right)
                NextLevel+=1
            if ToBePrint ==0: #该层的节点数已经打印结束
                res_val.append(temp)
                
                ToBePrint = NextLevel  #更新变量
                NextLevel = 0
                temp = []
        return res_val

提交结果:


—————————————————————————————————
题目三:
之字型打印二叉树

思路:
这道题可以受上面题目二的启发,即在上述的条件下,再增加一个lefttoright的变量,初始化为true,一层打印结束之后,将lefttoright反转,当lefttoright到为真时,和上述一致,当为假时,将temp倒叙加入到列表中,具体如下:

32 之字形打印二叉树.png

代码实现:

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回从上到下每个节点值列表,例:[1,2,3]
    def Print(self, pRoot):
        # write code here
        if pRoot ==  None:
            return []
        res = []
        res_val = []
        res.append(pRoot)
        leftToright = True

        ToBePrint = 1 #根节点为一个节点
        NextLevel = 0 #下一层的节点数
        temp = []     #储存每层节点的变量 
        while len(res)>0:
            
            node = res.pop(0)
            temp.append(node.val)
            ToBePrint -=1    #要打印的数目减1

            if node.left:
                res.append(node.left)
                NextLevel+=1  #x下一层节点数加1
            if node.right:
                res.append(node.right)
                NextLevel+=1
            if ToBePrint ==0: #该层的节点数已经打印结束
                if leftToright:
                    res_val.append(temp)
                else:
                    res_val.append(temp[::-1])
                ToBePrint = NextLevel  #更新变量
                NextLevel = 0
                temp = []
                leftToright = not leftToright
        return res_val

提交结果:

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,047评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,807评论 3 386
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,501评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,839评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,951评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,117评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,188评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,929评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,372评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,679评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,837评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,536评论 4 335
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,168评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,886评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,129评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,665评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,739评论 2 351

推荐阅读更多精彩内容