列表字典作为树类

Question

我正在尝试定义一个易于管理的树类（添加节点、获取子节点并以 dfs 或 bfs 样式遍历它），但我对 python 了解一点，运行时遇到错误我的代码：

AttributeError: 'Tree' 对象没有属性 'get'

这是我的代码，现在我刚刚实现了节点的添加：

class Tree(dict):
    def __init__(self):
        self = {}

    def add(self, node, child):
        if self.get(int(node)):
            childs = self.get(int(node))
            childs.append(int(child))
            self.update({int(node):childs})
        else:
            self.update({int(int(node)):[int(child)]})

def main():
    tot = int(input("Cantidad de relaciones: "))
    t = Tree()
    for i in range (0, tot):
        for i in range (0,1):
            a = []
            a.extend(input().split())
        t.add(a[0], a[1])
    print(t)

main()

Answer 1

从dict继承它！

get 和update 是字典数据类型的方法。

您的代码：class Tree(object): 将其替换为：class Tree(dict):，错误将得到解决。

您对代码的想法也很奇怪。 Python 已经实现了具有dict 和list 数据类型的树状数据结构。

a = {
    'a': [1, 2, 3],
    'b': {
        'c': [33, 44, 55]
        'd': 123,
    }
}

Python data structures