剑指offer、把二叉树打印成多行(python)

题目:

  • 从上到下按层打印二叉树,同一层节点从左向右输出,每一层输出一行

代码(python):

# -*- 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 pRoot == None:    # 若二叉树为空,则返回空列表
            return []
        
        queue1 = [pRoot]     # 定义队列1,把根结点赋值给他
        queue2 = []          # 定义队列2
        ret = []     # 最终的结果
        while queue1 or queue2:   # 若两个队列其中一个不空,则一直循环:
            if queue1:          # 若队列1存在
                tmpret = []        # 临时结果列表
                while queue1:    # 遍历队列1的所以元素
                    tmpNode = queue1[0]     # 将头元素,赋值给tmpNode
                    tmpret.append(tmpNode.val)   # 添加到临时结果列表
                    del queue1[0]      # 删除头元素
                    if tmpNode.left:       # 若tmpNode存在左节点
                        queue2.append(tmpNode.left)    # 则将左节点赋值给队列2
                    if tmpNode.right:      # 若tmpNode存在右节点
                        queue2.append(tmpNode.right)   # 则将右节点赋值给队列2
                ret.append(tmpret)      # 添加到结果列表
            if queue2:         # 若队列2存在
                tmpret = []      # 临时结果列表
                while queue2:     # 遍历队列2的所以元素
                    tmpNode = queue2[0]     # 将头元素,赋值给tmpNode
                    tmpret.append(tmpNode.val)    # 添加到临时结果列表
                    del queue2[0]    # 删除头元素
                    if tmpNode.left:   # 若tmpNode存在左节点
                        queue1.append(tmpNode.left)    # 则将左节点赋值给队列1
                    if tmpNode.right:      # 若tmpNode存在右节点
                        queue1.append(tmpNode.right)   # 则将右节点赋值给队列1
                ret.append(tmpret)
        return ret
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值