使用树的根将树存储为字典

Question

我有一个根节点，具有属性数据、父节点和子节点（列表）。我想使用一种方法递归地将整个树存储在字典中。

treeNode类

class treeNode:
    is_leaf = True
    root = False
    
    def __init__(self,data):
        self.data = data
        self.parent = None
        self.children = []

    def __repr__(self):
        return f"{self.data}"
    
    def add_child(self,child):
        child.parent = self
        self.children.append(child)
        self.is_leaf = False
    
    def get_parent(self):
        if self.parent is not None:
            return self.parent
        else:
            self.root = True
            return None
    
    def get_children(self):
        if not self.is_leaf:
            return self.children
        else:
            print("This is a leaf")

    def set_frequency(self,freq):
        self.frequency = freq

    ##This is the method I want to use
    ## First input (node) is root
    def get_all_tree(self,node):       
        if node.is_leaf:
            children.append(node)
        else:
            for x in node.get_children():
                self.get_all_tree(x) 

我该如何实现这一点？

编辑：我想创建一个字典树，有点像这样：

{0:[1，{2:[11,12,13,16，{3:[14,15]}，{4:[41,42,43]}]}

Answer 1

class treeNode:
...
    def get_all_tree(self, node):
        if node.is_leaf:
            return node
        else:
            tree = {node: []}
            for x in node.get_children():
                tree[node].append(x.get_all_tree(x))
            return tree

root = treeNode('0')
child1 = treeNode('1')
child2 = treeNode('2')
root.add_child(child1)
root.add_child(child2)
child1_1 = treeNode('1.1')
child1_2 = treeNode('1.2')
child1.add_child(child1_1)
child1.add_child(child1_2)
child1_1_1 = treeNode('1.1.1')
child1_1.add_child(child1_1_1)
print(root.get_all_tree(root))

这将返回{0:[{1:[{1.1:[1.1.1]}，1.2]}，2]}

我相信你的代码中还有一些改进，正如你所看到的，添加子元素是非常乏味的。考虑到类名为treeNode我相信您也有一个Tree类，我建议在Tree类中移动对子节点和父节点的处理，但将有关每个节点父节点/子节点的信息保留在treeNode类中。稍后我会尝试用一个简单的例子来解释它。

到目前为止，你的实施工作做得很好！