Python的图形实现问题

Question

这是我的节点类

class Node:
    def __init__(self, id, value):
        self.id = id
        self.value = value
        self.neighbors = set()

    def add_edge(self, node):
        self.neighbors.add(node)

    def get_adjacent_vertices(self):
        return sorted(list(self.neighbors))

    def __eq__(self, other):
        return self.id == other.id

    def __lt__(self, other):
        return self.id < other.id

    def __gt__(self, other):
        return self.id > other.id

    def __ge__(self, other):
        return self.id >= other.id

    def __le__(self, other):
        return self.id <= other.id

这是图形对象

class Graph:
    def __init__(self, directed=False):
        self.vertices = 0
        self.directed = directed
        self.vertex_map = {}

    def add_edge(self, v1, v2, weight=1):
        if v1 in self.vertex_map:
            V1 = self.vertex_map[v1]
        else:
            self.vertices += 1
            V1 = Node(self.vertices, v1)
            self.vertex_map[v1] = V1
        if v2 in self.vertex_map:
            V2 = self.vertex_map[v2]
        else:
            self.vertices += 1
            V2 = Node(self.vertices, v2)
            self.vertex_map[v2] = V2
        if V1 != V2:
            V1.add_edge(V2)
            if not self.directed:
                V2.add_edge(V1)
        else:
            raise GraphError("Cannot add node to itself!")

这是我的电话号码

if __name__ == '__main__':
    graph = Graph(directed=True)
    graph.add_edge('A', 'B')
    graph.add_edge('A', 'C')
    graph.add_edge('B', 'D')
    graph.add_edge('D', 'E')
    print(graph)

这会引发以下错误

TypeError: unhashable type: 'Node'

我可以通过将邻居更改为一个列表来解决这个问题，但是我的问题是，既然我已经为一个节点定义了哈希逻辑，那么如何继续使用集合来存储它的邻居呢

Answer 1

我认为你可以使用每个元素的id，它对每个元素都是唯一的。也可以使用defaultdict来代替set，它将用于存储顶点的所有相邻元素

从集合导入defaultdict 相邻节点\u list=defaultdict（lambda:[]）

def add_edge(self,u,v):
    if v not in d[u]:
        d[u].append(v)

如果这个图是无向的，那么你也必须把u加到d[v]上 如需更多输入，请指定您实际尝试的操作