分行从上往下打印二叉树

采用BFS广度优先搜索,关键是利用队列维护正在访问层的所有节点。

目录

一、题目

二、解答


一、题目

二、解答

利用python的 deque 构建队列。

思路:首先将根节点加入队列,利用 size(deque) 维护当前访问层的节点数,将本层的节点加入列表,然后pop已经访问过的节点,将其左右子树节点加入队列,循环往复直至deque队列中无访问节点。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
from collections import deque
class Solution(object):
    def printFromTopToBottom(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if not root:
                return []
        
        result = []
        queue = deque([root]) #根节点入队列
    
        while queue:
            level = []
            size = len(queue)  # 当前层的节点数
    
            for _ in range(size):
                node = queue.popleft() #pop
                level.append(node.val) #加入列表
    
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
    
            result.append(level)
    
        return result

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值