如何将数据添加到树？

Question

对于光束搜索，我创建了这棵树：

class Tree:

    def __init__(self, data):
        self.data = data
        self.children = []

    def add_child(self, data):
        child = Tree(data)
        self.children.append(child)

所以如果我想为n个级别创建k个孩子，那么如何填充这棵树？或任何更好的光束搜索解决方案？

我创建了这个用于添加数据，但是它只能添加一个孩子

def create_tree(t, k, child=None):

    k -= 1

    if k==0:
        return t
    else:

        if child == None:
            child = t

        child.add_child(1)
        return create_tree(t, k, child=t.children[0])