构建二叉树 python

#结点类
class Node():
    def __init__(self, item):
        self.item = item
        self.left = None
        self.right = None
#构建树
class Tree():
    def __init__(self):
        self.root = None
    
    #添加节点
    def add(self, item):
        node = Node(item)
        #根
        if self.root is None:
            self.root = node
        #用序列模拟对列,将根先加入对列
        else:
            q = [self.root]
            while q:
                #弹出队首
                point = q.pop(0)
                if point.left is None:
                    point.left = node
                    return 
                if point.right is None:
                    point.right = node
                    return 
                else:
                    q.append(point.left)
                    q.append(point.right)
    
    #层序遍历
    def travel(self, root):
        if root is None:
            return []
        l = []
        q = []
        q.append(root)
        while q:
            point = q.pop(0)
            l.append(point.item)
            if point.left:
                q.append(point.left)
            if point.right:
                q.append(point.right)
        return l

#测试
t = Tree()
for i in [1,2,3,4,5,6,7,8,9]:
    t.add(i)
print(t.travel(t.root))
				
#结果
[1,2,3,4,5,6,7,8,9]
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值