Python - 字典到NodeTree

时间:2012-11-27 00:13:09

标签: python python-2.7

我向你征求意见 我有一个NodeTree类。 然后我有一个函数create_T(d) 此函数将表示树的字典d作为输入,并创建具有NodeTree类型的节点的树 你建议我做什么? 感谢所有

1 个答案:

答案 0 :(得分:0)

您的信息很少,所以我做了一些假设。基本上我认为你需要知道从树开始的位置。如果你选择一个随机点,那么你必须向上和向下工作。如果你可以解决哪个是树的顶部,那么你只需要向下工作。我通过进行集合比较来计算树的顶部,以查看哪个节点不是任何其他节点的子节点。

import itertools

# NodeTree
class Node(object):
    def __init__(self, name):
        self.name = name
        self.children = []

    def add_child(self, child_node):
        self.children.append(child_node)

    def __repr__(self):
        if not self.children:
            return '<%s/>' % self.name
        else:
            children = ''.join([repr(child) for child in self.children])
            return '<%s>%s</%s>' % (self.name, children, self.name)

# create_T(d)
def create_tree(tree_dict):
    # find which node is the top of the tree (assume it is well-formed)
    all_leaves = set(itertools.chain(*tree_dict.values()))
    top_of_the_tree = set(tree_dict.keys()) - all_leaves

    def get_node(name):
        node = Node(name)
        for child_name in tree_dict[name]:
            node.add_child(get_node(child_name))
        return node
    return get_node(top_of_the_tree.pop())

print create_tree({'a': ['b', 'c'],
                   'b': ['d'],
                   'c': ['e'],
                   'd': [],
                   'e': [],
                   'f': ['a']})

,并提供:

   <f><a><b><d/></b><c><e/></c></a></f>