python - 构建一棵树 "backwards"

Question

我需要在 Python 中创建一个树状结构。我有一个函数 get(parentId)，它返回具有该父级的对象列表——我认为应该递归地完成。

结果应该是这样的:["root object", ["child1 of root", "child2 of root", ["child2-1", "child2-2"]]]

每个对象都有一个属性parent，它是get()的parentId，但作为起点，我只有根对象。

Answer 1

假设您仍然对树的列表表示感兴趣(这不一定是无用的事情)，这是一个递归函数定义，我相信它可以满足您的需要(前提是 get() 函数确实之前已定义):

def build_tree(node):
    return [node,[build_tree(child) for child in get(node)]]

您可以通过类似于以下的方式使用它:

root = 1  # or whatever other representation you may use for root
list = build_tree(root)
print list