我的代码首先接受用户输入:
n
=图表中的顶点数
file
=要计算的文件
在这种情况下,我的文件是一个包含以下内容的.txt文件:
0 1
2 1
0 2
1 3
我的代码成功将其转换为输出[[1, 2], [0, 2, 3], [1, 0], [1]]
的邻接列表。然而,问题是将其转换为邻接矩阵。我知道我的代码的主要问题是在这一行:
for x in range(len(vertices)):
matrix[z-1][vertices[x]] = 1
这是我的完整代码:
n = int(input("Enter the number of vertices: ")) ## E.g. 4
file = input("Enter the filename: ") ## E.g. graph.txt
vertices = []
matrix = [] ## define list
for x in range(n):
matrix.append([0]*n]) ## append a list for n, e.g. if n = 4 then [[][][][]]
vertices.append([])
f =open(file)
z = 0
for line in f: ##loop statement until no more line in file
line = line.split() ## turn every line into a list
z+=1
for y in range(len(line)):
line[y] = line[y].strip() ## remove spaces
line[y] = int(line[y]) ## converts line list into integer
## add value of line into main list
vertices[line[0]].append(line[1])
vertices[line[1]].append(line[0])
for z in range(len(matrix)):
for x in range(len(vertices)):
matrix[z-1][vertices[x]] = 1
print(vertices)
print(matrix)
答案 0 :(得分:0)
首先,您使用z-1
作为索引,可以为-1,因此应该只是z
。
其次,您在内部循环中访问vertices[x]
,就好像它有一个节点号一样,但它是一个嵌套列表。相反,您需要vertices[z][x]
。为了提高性能,原则上最好为vertices[z]
创建一个名称:
for z in range(len(matrix)):
neighbors = vertices[z]
for x in range(len(neighbors)):
matrix[z][neighbors[x]] = 1
或者:
for z, neighbors in enumerate(vertices):
for x in neighbors:
matrix[z][x] = 1