Python程序查找二叉树中所有节点的总和

在需要查找树的所有节点的总和时，可以创建一个类并包含设置根节点，向树中添加元素，搜索特定元素，添加树元素以查找总和等方法。可以创建类的实例以访问和使用这些方法。

以下是相同操作的演示 –

更多Python相关文章，请阅读：Python 教程

例子

class Tree_struct:
    def __init__(self, data=None):
        self.key = data
        self.children = []

    def set_root(self, data):
        self.key = data

    def add_node(self, node):
        self.children.append(node)

    def search_node(self, key):
        if self.key == key:
            return self
        for child in self.children:
            temp = child.search_node(key)
            if temp is not None:
                return temp
        return None

    def sum_node(self):
        my_summation = self.key
        for child in self.children:
            my_summation = my_summation + child.sum_node()
        return my_summation

my_instance = None

print('菜单（假定没有重复的键）')
print('add <data> at root')
print('add <data> below <data>')
print('sum')
print('quit')

while True:
    my_input = input('您要执行什么操作？').split()

    operation = my_input[0].strip().lower()
    if operation == 'add':
        data = int(my_input[1])
        new_node = Tree_struct(data)
        suboperation = my_input[2].strip().lower()
        if suboperation == 'at':
            my_instance = new_node
        elif suboperation == 'below':
            position = my_input[3].strip().lower()
            key = int(position)
            ref_node = None
            if my_instance is not None:
                ref_node = my_instance.search_node(key)
            if ref_node is None:
                print('没有这个键')
                continue
            ref_node.add_node(new_node)

    elif operation == 'sum':
        if my_instance is None:
            print('树为空')
        else:
            my_summation = my_instance.sum_node()
            print('所有节点的总和为：{}'.format(my_summation))

    elif operation == 'quit':
        break

输出

菜单（假定没有重复的键）
add <data> at root
add <data> below <data>
sum
quit
您要执行什么操作？add 5 at root
您要执行什么操作？add 7 below 5
您要执行什么操作？add 0 below 7
您要执行什么操作？sum
所有节点的总和为：12
您要执行什么操作？quit

说明

• 创建了“Tree_struct”类并具有所需的属性。

• 它有一个“init”函数，用于创建一个空列表。

• 定义了一个名为’set_root’的方法，它有助于设置二叉树的根值。

• 它有一个“add_node”方法，它帮助将元素添加到树中。

• 定义了一个名为’search_elem’的方法，它有助于搜索特定元素。

• 定义了一个名为’sum_node’的方法，它有助于添加树元素并找到总和。

• 创建了一个实例并分配给“None”。

• 获取用户输入的执行操作。

• 根据用户的选择，执行操作。

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