使用二叉搜索树排序的Python程序


当需要对二叉搜索树进行排序时,会创建一个类,并在其中定义执行插入元素和执行中序遍历等操作的方法。

下面是相同的演示 -

示例

class BinSearchTreeNode:
   def __init__(self, key):
      self.key = key
      self.left = None
      self.right = None
      self.parent = None
   def insert_elem(self, node):
      if self.key > node.key:
         if self.left is None:
            self.left = node
            node.parent = self
         else:
            self.left.insert_elem(node)
      elif self.key <= node.key:
         if self.right is None:
            self.right = node
            node.parent = self
         else:
            self.right.insert_elem(node)

   def inorder_traversal(self):
      if self.left is not None:
         self.left.inorder_traversal()
      print(self.key, end=' ')
      if self.right is not None:
         self.right.inorder_traversal()

class BinSearchTree:
   def __init__(self):
      self.root = None

   def inorder_traversal(self):
      if self.root is not None:
         self.root.inorder_traversal()

   def add_val(self, key):
      new_node = BinSearchTreeNode(key)
      if self.root is None:
         self.root = new_node
      else:
         self.root.insert_elem(new_node)

my_instance = BinSearchTree()

my_list = input('Enter the list of numbers... ').split()
my_list = [int(x) for x in my_list]
for x in my_list:
   my_instance.add_val(x)
print('Sorted list: ')
print(my_instance.inorder_traversal())

输出

Enter the list of numbers... 67 54 89 0 11 34 99
Sorted list:
0 11 34 54 67 89 99

解释

  • 创建具有所需属性的“BinSearchTreeNode”类。

  • 它有一个“init”函数,用于将“left”、“right”和“parent”节点赋值为“None”。

  • 定义另一个名为“insert_elem”的方法,该方法有助于将节点添加到树中。

  • 定义另一个名为“inorder_traversal”的方法,该方法有助于对树执行中序遍历。

  • 定义另一个名为“BinSearchTree”的类。

  • 它将根节点设置为“None”。

  • 它有一个名为“inorder_traversal”的方法,该方法有助于对树执行中序遍历。

  • 定义另一个名为“add_val”的方法,该方法有助于将节点添加到树中。

  • 创建一个“BinSearchTree”的实例。

  • 用户输入数字列表。

  • 使用此列表创建一个树。

  • 对列表进行排序,并执行其中序遍历。

  • 在控制台上显示相关的输出。

更新于:2021年4月16日

693 次浏览

开启您的职业生涯

通过完成课程获得认证

开始
广告