修复Python中的深度树

时间:2010-10-21 08:12:08

标签: python data-structures tree

我想实现一个具有固定深度的树结构,即当将子项添加到leef节点时,整个树结构应该“向上移动”。这也意味着可以同时存在多个根。见下面的示例: alt text 在此示例中,绿色节点在迭代1中添加,删除顶部节点(灰色)并使两个蓝色节点在K = 0和迭代1根节点。

我该如何实施?

1 个答案:

答案 0 :(得分:2)

为每个节点存储对其父节点的引用。当您将一个节点作为子节点添加到它时,向上走父节点(从添加到的节点)并在将其所有子节点中的父引用设置为None后删除第三个节点。然后将已删除节点的子节点添加到树列表中。

class Node(object):

    depth = 4

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


def create_node(trees, parent, contents):
    """Adds a leaf to a specified node in the set of trees.

    Note that it has to have access to the container that holds all of the trees so
    that it can delete the appropriate parent node and add its children as independent 
    trees. Passing it in seems a little ugly. The container of trees could be a class
    with this as a method or you could use a global list. Or something completely
    different. The important thing is that if you don't delete every reference to the
    old root, you'll leak memory.
    """
    parent.children.append(Node(parent, contents))

    i = 0:
    L = Node.depth - 1
    while i < L:
        parent = parent.parent
        if not parent:
            break
        i += 1
    else:
        for node in parent.children:
            node.parent = None
        trees.extend(parent.children)
        i = trees.find(parent)
        del trees[i]