python邻接矩阵实现图类
时间: 2024-04-17 19:20:46 浏览: 111
python将邻接矩阵输出成图的实现
Python中可以使用邻接矩阵来实现图类。邻接矩阵是一个二维数组,用于表示图中各个节点之间的连接关系。在邻接矩阵中,行和列分别代表图中的节点,矩阵中的元素表示节点之间的连接情况。
下面是一个简单的Python代码示例,展示了如何使用邻接矩阵实现图类:
```python
class Graph:
def __init__(self, num_vertices):
self.num_vertices = num_vertices
self.adj_matrix = [[0] * num_vertices for _ in range(num_vertices)]
def add_edge(self, v1, v2):
if v1 >= 0 and v1 < self.num_vertices and v2 >= 0 and v2 < self.num_vertices:
self.adj_matrix[v1][v2] = 1
self.adj_matrix[v2][v1] = 1
def remove_edge(self, v1, v2):
if v1 >= 0 and v1 < self.num_vertices and v2 >= 0 and v2 < self.num_vertices:
self.adj_matrix[v1][v2] = 0
self.adj_matrix[v2][v1] = 0
def print_graph(self):
for row in self.adj_matrix:
print(row)
# 创建一个包含5个节点的图
graph = Graph(5)
# 添加边
graph.add_edge(0, 1)
graph.add_edge(0, 4)
graph.add_edge(1, 2)
graph.add_edge(1, 3)
graph.add_edge(1, 4)
graph.add_edge(2, 3)
graph.add_edge(3, 4)
# 打印图的邻接矩阵
graph.print_graph()
```
这段代码创建了一个包含5个节点的图,并添加了一些边。通过调用`print_graph`方法,可以打印出图的邻接矩阵。
阅读全文