R向图的邻接矩阵添加名称属性

class Graph:
    def __init__(self, vertices):
        self.V = vertices
        self.graph = [[0 for column in range(vertices)] for row in range(vertices)]
        self.vertex_names = {}

    def add_vertex_name(self, vertex, name):
        self.vertex_names[vertex] = name

    def add_edge(self, u, v, weight=1):
        self.graph[u][v] = weight

    def print_graph(self):
        for i in range(self.V):
            for j in range(self.V):
                print(self.graph[i][j], end=" ")
            print(" -> ", self.vertex_names.get(i, "None"))

# 创建一个有向图
g = Graph(4)
g.add_vertex_name(0, 'A')
g.add_vertex_name(1, 'B')
g.add_vertex_name(2, 'C')
g.add_vertex_name(3, 'D')

# 添加边
g.add_edge(0, 1)
g.add_edge(0, 2)
g.add_edge(1, 2)
g.add_edge(2, 0)
g.add_edge(2, 3)
g.add_edge(3, 3)

# 打印图
g.print_graph()