Question

这是我的Node类，

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'

我可以通过将邻居更改为列表来解决此问题，但是我的问题是，因为我已经为Node定义了哈希算法，所以如何继续使用集合存储邻居。

Answer 1

好吧，我认为您可以为每个元素使用ID，这对于每个元素都是唯一的。除了设置之外，您还可以使用defaultdict来存储顶点的所有相邻元素。

从集合

导入defaultdict neighbor_node_list = defaultdict（lambda：[]）

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

如果图形是无向的，则还必须将u添加到d [v] 如需更多输入，请指定您实际要执行的操作

图实现问题是Python

1 个答案: