Python - Dictionary to NodeTree

Question

조언을 구합니다. NodeTree 클래스가 있습니다. 다음 create_T (d) 함수가 있습니다 이 함수는 트리를 나타내는 사전 d를 입력으로 받아 NodeTree 유형의 노드가있는 트리를 만듭니다. 어떻게해야합니까? 감사합니다 모두

Answer 1

정보가 부족하여 몇 가지 가정을했습니다. 기본적으로 당신은 나무에서 어디서부터 시작할지를 알아야한다고 생각합니다. 임의의 지점을 선택하면 위쪽 및 아래쪽으로 작업해야합니다. 당신이 나무의 꼭대기 인 것을 운동 할 수있는 경우에 당신은 다만 아래쪽으로 일할 필요가있다. 어떤 노드가 다른 노드의 하위 노드가 아닌지 확인하기 위해 집합 비교를 수행하여 트리의 맨 위를 계산했습니다.

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>