写一段Python的图类数据结构代码
时间: 2023-11-21 15:04:52 浏览: 31
好的,以下是一个简单的Python图类数据结构代码:
```python
class Graph:
def __init__(self, vertices):
self.vertices = set(vertices)
self.edges = dict((v, []) for v in vertices)
def add_edge(self, u, v):
self.edges[u].append(v)
self.edges[v].append(u)
def remove_edge(self, u, v):
self.edges[u].remove(v)
self.edges[v].remove(u)
def get_neighbors(self, v):
return self.edges[v]
def is_connected(self, u, v):
return u in self.edges[v]
```
注释:
- `Graph` 是一个简单的图类,它包含操作图的常用方法。
- `__init__` 方法初始化图的顶点和边,其中 `vertices` 是一组顶点的集合,`edges` 是一个字典,键为每个顶点,值为这个顶点的邻接列表。
- `add_edge` 方法添加一条边,即将 `u` 和 `v` 两个顶点相连。
- `remove_edge` 方法删除一条边,即将 `u` 和 `v` 两个顶点断开连接。
- `get_neighbors` 方法获取一个顶点的邻居顶点列表。
- `is_connected` 方法判断两个顶点是否相连。