python二叉树的层序遍历模版

1. BFS使用队列

使用两个队列存储二叉树,主队列循环遍历各层,临时队列对每一层元素遍历,代码如下,res存放层序遍历结果:

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return []
        queue = [root]      #主队列
        res = []
        level = 0		#标记层号
        while len(queue) > 0:
            res.append([])
            temp = []       #临时队列
            for p in queue:
                res[level].append(p.val)
                if p.left:
                    temp.append(p.left)
                if p.right:
                    temp.append(p.right)
            queue = temp
            level += 1
        return res

时间复杂度:O(N)
空间复杂度:O(N)

2. DFS

class Solution:
    def dfs(self, root, d, res):
        if not root:
            return
        if len(res) == d:
            res.append([])
        res[d].append(root.val)
        self.dfs(root.left, d+1, res)
        self.dfs(root.right, d+1, res)

    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return []
        res = []
        self.dfs(root, 0,  res)
        return res

时间复杂度:O(N)
空间复杂度:O(N)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值