队列+广度优先搜索(bfs)剑指 Offer 32 . 从上到下打印二叉树

python当中collections有一个deque双向队列,其popleft()可以实现从队列中出时间复杂度为o(1)

剑指 Offer 32 - I. 从上到下打印二叉树

直接使用队列,初始队列中为root,为从左到右便利则依次往队列中添加二叉树的结点即可,当队列为空则证明遍历结束

加一个结点利用popleft()队列出一个

import collections
class Solution:
    def levelOrder(self, root: TreeNode) -> List[int]:
        if not root: return []
        dq = collections.deque()
        dq.append(root)
        res = []
        while dq:
            node = dq.popleft()
            res.append(node.val)
            if node.left: dq.append(node.left)
            if node.right: dq.append(node.right)
        return res

剑指 Offer 32 - II. 从上到下打印二叉树 II

import collections
class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:return []
        dq = collections.deque()
        res = []
        dq.append(root)
        while dq:
            temp = []
            for _ in range(len(dq)):
                i = dq.popleft()
                temp.append(i.val)
                if i.left: dq.append(i.left)
                if i.right:dq.append(i.right)
            res.append(temp)
        return res

依旧使用上一题的队列,但是需要修改一个循环内的队列条件,实现逐层添加到结果,此题关键点就是使用当前层的队列长度,之后只遍历这段长度,向队列后加结点即可。

剑指 Offer 32 - III. 从上到下打印二叉树 III

上一题直接讲temp数组翻转即可

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值