python实现二叉树层次遍历(BFS)


# 求二叉树路径DFS
class TreeNode:
    def __init__(self, val = None):
        self.left = None
        self.right = None
        self.val = val
class BTree:
    def insert(self,root,node): #构建二叉排序树
        if root is None:
            return node
        if root.val < node.val:
            root.right = self.insert(root.right,node)
        else:
            root.left = self.insert(root.left,node)
        return root
    def order_print(self,root):
        if root is None:
            return None
        self.order_print(root.left)
        print(root.val, end=" ")
        self.order_print(root.right)
    def dfs(self, root, path, path_lst):
        if root:
            path += str(root.val) # 每递归到一个节点就加入该path
            if not root.left and not root.right: # 遇到叶子节点则把整个path加入path_lst
                path_lst.append(path)
            else:
                path += '->'
                self.dfs(root.left,path, path_lst)
                self.dfs(root.right,path, path_lst)
        return path_lst
    def bfs(self, root):
        level_order = []
        que = [root]
        if root == None:
            return level_order
        while que:
            templist = []
            for i in range(len(que)):
                node = que.pop(0)
                templist.append(node.val)
                if node.left:
                    que.append(node.left)
                if node.right:
                    que.append(node.right)
            level_order.append(templist)
        return level_order

lst = [5,1,2,3,6,8,9]
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.left = TreeNode(6)
root.right.right = TreeNode(7)
bt = BTree()
# for i in range(1, len(lst)):
#     bt.insert(root, TreeNode(lst[i]))
bt.order_print(root)
print()
print(bt.bfs(root))

在这里插入图片描述

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Shashank497

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值