计算给定树的非叶节点数的 Python 程序

  • Z时代
  • 2024-01-10
  • 分类：综合

当需要查找树中非叶节点的数量时，会创建一个“Tree_structure”类，定义设置根值和添加其他值的方法。给出了用户可以选择的各种选项。根据用户的选择，对Tree元素进行操作。

以下是相同的演示 -

示例

class Tree_structure:
   def __init__(self, data=None):
     self.key= data
     self.children= []
   def set_root(self, data):
     self.key= data
   def add_vals(self, node):
      self.children.append(node)
   def search_val(self, key):
      ifself.key== key:
         return self
      for child in self.children:
         temp = child.search_val(key)
         if temp is not None:
            return temp
      return None
   def count_non_leaf_node(self):
      nonleaf_count = 0
      ifself.children!= []:
         nonleaf_count = 1
      for child in self.children:
         nonleaf_count = nonleaf_count + child.count_non_leaf_node()
      return nonleaf_count
tree = None
print('Menu (this assumes no duplicate keys)')
print('add <data> at root')
print('add <data> below <data>')
print('count')
print('quit')
while True:
   my_input = input('What operation would you like to perform ? ').split()
   operation = my_input[0].strip().lower()
   if operation == 'add':
      data = int(my_input[1])
      newNode = Tree_structure(data)
      suboperation = my_input[2].strip().lower()
      if suboperation == 'at':
         tree = newNode
      elif suboperation == 'below':
         position = my_input[3].strip().lower()
         key = int(position)
         ref_node = None
         if tree is not None:
            ref_node = tree.search_val(key)
         if ref_node is None:
            print('No such key.')
            continue
         ref_node.add_vals(newNode)
   elif operation == 'count':
      if tree is None:
         print('The tree is empty ')
      else:
         count = tree.count_non_leaf_node()
         print('The number of non-leaf nodes are : {}'.format(count))
   elif operation == 'quit':
      break
输出结果
Menu (this assumes no duplicate keys)
add <data> at root
add <data> below <data>
count
quit
What operation would you like to perform ? add 34 at root
What operation would you like to perform ? add 78 below 34
What operation would you like to perform ? add 56 below 78
What operation would you like to perform ? add 90 below 56
What operation would you like to perform ? count
The number of non-leaf nodes are : 3
What operation would you like to perform ? quit

解释

  • 'Tree_structure' 类被创建。

  • 它将“键”设置为 True 并将一个空列表设置为树的子项。

  • 它有一个“set_root”函数，可以帮助设置树的根值。

  • 定义了一个名为“add_vals”的方法，它有助于将元素添加到树中。

  • 定义了另一个名为“search_val”的方法，它有助于搜索树中的元素。

  • 定义了另一个名为“count_non_leaf_nodes”的方法，它有助于获取树的非叶节点的计数。

  • 它是一个递归函数。

  • 给出了四个选项，例如“在根目录添加”、“在下面添加”、“计数”和“退出”。

  • 根据用户给出的选项，执行相应的操作。

  • 此输出显示在控制台上。

以上是 计算给定树的非叶节点数的 Python 程序 的全部内容， 来源链接： utcz.com/z/335578.html

回到顶部