строить дерево в python

Question

я пытаюсь построить дерево в соответствии с заданным списком элементов, где каждый элемент составляет целый уровень. element0 - это root, element1 - первый уровень и так далее.

это мой код:

example: for a given list ["fever", "headache", "fatigue"] my tree would be:
                    fever
    #          Yes /       \ No
    #        headache           headache
    #   Yes /     \ No     Yes/     \No
    #     fatigue   fatigue  fatigue fatigue 
class Node:
    def __init__(self, data, positive_child = None, negative_child = None):
        self.data = data
        self.positive_child = positive_child
        self.negative_child = negative_child
    def __str__(self):
        return str(self.data)

def tree_helper(symptoms,tree_nodes):
    if len(symptoms) == 0:
        return
    if len(symptoms) == 1:
        symptoms[0] = Node(symptoms[0])
        tree_nodes.append(symptoms[0])
        print('OK')
    else:
        yes_root = Node(symptoms[0],symptoms[1],symptoms[1])

        tree_nodes.append((yes_root))
        tree_helper(symptoms[1:],tree_nodes)
        no_root = Node(symptoms[0],symptoms[1],symptoms[1])
        tree_nodes.append(no_root)
        tree_helper(symptoms[1:],tree_nodes)
        return tree_nodes

def build_tree(symptoms):
    tree_nodes = []
    if len(symptoms) > 1:
        tree_nodes = tree_helper(symptoms,tree_nodes)
    for i in tree_nodes:
        print(i.data, i.positive_child,i.negative_child)

if __name__ == "__main__":
    build_tree(["fever","headache","fatigue"])

кажется, что он создает узлы только с самого левого пути и самого правого пути и добавляет их дважды.

Answer 1

Вы можете построить узлы, а затем добавить соединения.

# each layer has 2^i nodes for i = 0,1,2..
nodes = [[Node(s) for _ in range(pow(2,i))] for i,s in enumerate(s)]

# add connections
for i in range(len(nodes)-1):
    for parent in nodes[i]:
        parent.positive_child = nodes[i+1][2*i]
        parent.negative_child = nodes[i+1][2*i + 1]


tree = nodes[0][0]