class adjMatrixGraph: # 构造方法,n个顶点m条边 def init(self,n,m): self.verNum = n #顶点数 self.edgeNum = m #边数 self.vertex = [0] * n #顶点列表 self.edge = [[0 for i in range(self.verNum)] \ for j in range(self.verNum)] #邻接矩阵二维列表 self.vis = [False] * n #顶点的访问列表,默认没访问过 def addVertex(self,ls): #添加顶点列表 self.vertex = ls def addEdge(self,fr,to):#添加边(fr,to) ifr = self.vertex.index(fr) #起点下标 ito = self.vertex.index(to) #终点下标 self.edge[ifr][ito] = self.edge[ito][ifr] = 1 #邻接矩阵 #邻接矩阵建图 def createGraph(): n,m = map(int,input().split()) #输入n个顶点和m条边 g = adjMatrixGraph(n,m) #创建无向图G g.addVertex(list(input().split())) #输入顶点列表 for i in range(m): #输入m条边 fr,to = input().split() g.addEdge(fr,to) return g 根据上面代码编写一个广度优先搜索的函数
时间: 2024-03-20 19:39:59 浏览: 12
可以参考以下代码实现广度优先搜索:
```
from queue import Queue
class adjMatrixGraph:
# 构造方法,n个顶点m条边
def __init__(self, n, m):
self.verNum = n #顶点数
self.edgeNum = m #边数
self.vertex = [0] * n #顶点列表
self.edge = [[0 for i in range(self.verNum)] for j in range(self.verNum)] #邻接矩阵二维列表
self.vis = [False] * n #顶点的访问列表,默认没访问过
def addVertex(self, ls):
#添加顶点列表
self.vertex = ls
def addEdge(self, fr, to):
#添加边(fr,to)
ifr = self.vertex.index(fr) #起点下标
ito = self.vertex.index(to) #终点下标
self.edge[ifr][ito] = self.edge[ito][ifr] = 1 #邻接矩阵
# 广度优先搜索
def bfs(self, start):
visited = [False] * self.verNum # 标记所有节点是否被访问
q = Queue() # 创建队列
q.put(start) # 将起始节点入队
visited[self.vertex.index(start)] = True # 标记起始节点为已访问
while not q.empty(): # 队列非空
node = q.get() # 取队首节点
print(node, end=' ') # 输出节点值
# 遍历队首节点的所有邻居节点
for i in range(self.verNum):
if self.edge[self.vertex.index(node)][i] == 1 and not visited[i]:
visited[i] = True # 标记邻居节点为已访问
q.put(self.vertex[i]) # 将邻居节点入队
```
使用方式如下:
```
# 创建图对象
g = adjMatrixGraph(6, 7)
# 添加顶点列表
g.addVertex(['A', 'B', 'C', 'D', 'E', 'F'])
# 添加边
g.addEdge('A', 'B')
g.addEdge('A', 'C')
g.addEdge('B', 'D')
g.addEdge('C', 'D')
g.addEdge('C', 'E')
g.addEdge('D', 'E')
g.addEdge('D', 'F')
# 广度优先搜索
g.bfs('A')
```
输出结果为:`A B C D E F`