数据结构
网络
关系数据库管理系统
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C语言编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理学
化学
生物学
数学
英语
经济学
心理学
社会学
时尚研究
法律研究Python程序计算树中叶子节点的数量
当需要计算树中叶子节点的数量时，会创建一个名为“Tree_structure”的类，并定义添加根值和其他子值的方法。用户可以选择各种选项。根据用户的选择，对树元素执行操作。
下面是相同的演示 -
示例
class Tree_structure:
def __init__(self, data=None):
self.key = data
self.children = []
def set_root_node(self, data):
self.key = data
def add_vals(self, node):
self.children.append(node)
def search_val(self, key):
if self.key == key:
return self
for child in self.children:
temp = child.search(key)
if temp is not None:
return temp
return None
def count_leaf_node(self):
leaf_nodes = []
self.count_leaf_node_helper_fun(leaf_nodes)
return len(leaf_nodes)
def count_leaf_node_helper_fun(self, leaf_nodes):
if self.children == []:
leaf_nodes.append(self)
else:
for child in self.children:
child.count_leaf_node_helper_fun(leaf_nodes)
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)
sub_op = my_input[2].strip().lower()
if sub_op == 'at':
tree = newNode
elif sub_op == 'below':
my_pos = my_input[3].strip().lower()
key = int(my_pos)
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_leaf_node()
print('The number of 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 78 at root What operation would you like to perform ? add 90 below 78 What operation would you like to perform ? add 8 below 78 What operation would you like to perform ? count The number of leaf nodes are : 2 What operation would you like to perform ? quit
解释
创建“Tree_structure”类。
它将“key”设置为True，并将树的子节点设置为一个空列表。
它有一个名为“set_root”的函数，用于设置树的根值。
定义了一个名为“add_vals”的方法，用于向树中添加元素。
定义了一个名为“search_val”的方法，用于在树中搜索元素。
定义了一个名为“count_leaf_nodes”的方法，用于获取树中叶子节点的数量。
定义了一个名为“count_leaf_nodes_helper”的方法，它调用前面定义的函数 - 这是一个递归函数。
提供了四个选项，例如“添加到根节点”、“添加到下方”、“计数”和“退出”。
根据用户提供的选项，执行相应的操作。
此输出显示在控制台上。
更新于: 2021年4月15日
555 次查看
广告