实现二叉树的Python程序

当需要在Python中实现二叉树时,将使用面向对象的方法。在此,定义了一个类,并定义了属性。函数是在类中定义的,用于执行某些操作。将创建该类的实例,并将这些函数用于执行计算器操作。

以下是相同的演示-

示例

class binomial_tree:
   def __init__(self, key):
     self.key= key
     self.children= []
     self.order= 0
   def add_at_end(self, t):
      self.children.append(t)
     self.order=self.order+ 1
my_tree = []
print('Menu')
print('create <key>')
print('combine <index1> <index2>')
print('exit')
while True:
   option = input('What do you wish like to do? ').split()
   operation = option[0].strip().lower()
   if operation == 'create':
      key = int(option[1])
      b_tree = binomial_tree(key)
      my_tree.append(b_tree)
      print('Binomial tree has been created.')
   elif operation == 'combine':
      index_1 = int(option[1])
      index_2 = int(option[2])
      if my_tree[index_1].order == my_tree[index_2].order:
         my_tree[index_1].add_at_end(my_tree[index_2])
         del my_tree[index_2]
         print('Binomial trees have been combined.')
      else:
         print('Order of trees need to be the same to combine them.')
   elif operation == 'exit':
      print("Exit")
      break
   print('{:>8}{:>12}{:>8}'.format('Index', 'Root key', 'Order'))
   for index, t in enumerate(my_tree):
print('{:8d}{:12d}{:8d}'.format(index, t.key, t.order))
输出结果
Menu
create <key>
combine <index1> <index2>
exit
What do you wish like to do? create 7
Binomial tree has been created.
Index Root key Order
0 7 0
What do you wish like to do? create 11
Binomial tree has been created.
Index Root key Order
0 7 0
1 11 0
What do you wish like to do? create 4
Binomial tree has been created.
Index Root key Order
   0    7    0
   1    11    0
   2    4    0
What do you wish like to do? combine 0 1
Binomial trees have been combined.
Index Root key Order
   0    7    1
   1    4    0
What do you wish like to do? exit
Exit

解释

  • 定义了一个名为“ binomial_tree”的类。

  • 它具有将元素添加到树的末尾的方法。

  • 创建一个空列表。

  • 基于选项,用户选择一个选项。

  • 如果他们选择创建密钥,则将创建类的实例,并创建二叉树。

  • 还计算索引,根值和顺序。

  • 如果需要组合索引,则选择另一个选项,并且还提到需要组合的节点的索引值。

  • 这将合并数据并显示它。