二叉树简单实现

二叉树简单实现:

class Node: def __init__(self,item): self.item = item self.child1 = None self.child2 = 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 True: pop_node = q.pop(0) if pop_node.child1 is None: pop_node.child1 = node return elif pop_node.child2 is None: pop_node.child2 = node return else: q.append(pop_node.child1) q.append(pop_node.child2) def traverse(self): # 层次遍历 if self.root is None: return None q = [self.root] res = [self.root.item] while q != []: pop_node = q.pop(0) if pop_node.child1 is not None: q.append(pop_node.child1) res.append(pop_node.child1.item) if pop_node.child2 is not None: q.append(pop_node.child2) res.append(pop_node.child2.item) return res def preorder(self,root): # 先序遍历 if root is None: return [] result = [root.item] left_item = self.preorder(root.child1) right_item = self.preorder(root.child2) return result + left_item + right_item def inorder(self,root): # 中序序遍历 if root is None: return [] result = [root.item] left_item = self.inorder(root.child1) right_item = self.inorder(root.child2) return left_item + result + right_item def postorder(self,root): # 后序遍历 if root is None: return [] result = [root.item] left_item = self.postorder(root.child1) right_item = self.postorder(root.child2) return left_item + right_item + result t = Tree() for i in range(10): t.add(i) print('层序遍历:',t.traverse()) print('先序遍历:',t.preorder(t.root)) 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值