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>