《剑指offer》【把二叉树打印成多行】(python版)

题目描述:从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。

思路:本题实际上就是层序遍历,但是要点在于如何在做到不同行结点分开。这里考虑的是采用三个列表。保证nodelist只保存上一行结点。遍历nodelist中结点,将其左右孩子依次存入curnodelist中。遍历结束将curnodelist加入nodelist,并删除nodelist中第一个元素,即上一行结点。


class Solution:
    # 返回二维列表[[1,2],[4,5]]
    def Print(self, pRoot):
        # write code here
        nodelist = []
        outlist = []
        if pRoot:
        # nodelist中永远只有一个元素,这唯一的元素一定是一个包含某一层结点的列表
            nodelist.append([pRoot])
        else:
            return outlist
        while len(nodelist) and len(nodelist[0]):
            outlist.append([node.val for node in nodelist[0]])
            curnodelist = []
            # 遍历上一层结点,记录下一层结点,存入一个新的列表中
            for i in range(len(nodelist[0])):
                curnode = nodelist[0][i]
                if curnode.left:
                    curnodelist.append(curnode.left)
                if curnode.right:
                    curnodelist.append(curnode.right)
            # 更新保存上一层结点的列表
            nodelist.append(curnodelist)
            del nodelist[0]
        return outlist

测试二叉树结构:

![二叉树](https://img-blog.csdn.net/20180702175730219?watermark/2/text/aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzIwMTQxODY3/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70) 输出结果:
[[4], [2, 6], [3, 5, 7], [1]]
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值