给定两个房间的编号A和B,请找出从房间A到达房间B最少需要经过几个传送门。 例如:N=3,3个房间中传送门的编号分别为: 房间1:1、4、6; 房间2:2、3、4,8; 房间3:3、6、9。 其中房间1和房间2互通,共用4号传送门;房间1和房间3互通,共用6号传送门;房间2和房间3互通,共用3号传送门; 当A=1,B=2,从房间1到达房间2,共有两种路线: 路线1:从房间1通过4号传送门进入房间2,共经过1个传送门; 路线2:从房间1通过6号传送门进入房间3,再从房间3通过3号传送门进入房间2,共经过2个传送门; 故从房间1到达房间2最少需要经过1个传送门。 输入描述 第一行输入一个正整数N(2≤N≤20),表示房间数量 接下来输入N行,每行包含多个正整数(1≤正整数≤100),第2行到第N+1行依次表示1到N号房间内所有传送门的编号,正整数之间以一个英文逗号隔开 最后一行输入两个正整数A和B(1≤A≤N,1≤B≤N,且A≠B),表示两个房间的编号,正整数之间以一个英文逗号隔开 输出描述 输出一个整数,表示从房间A到达房间B最少需要经过几个传送门,如果房间A不能到达房间B,则输出-1
时间: 2023-11-20 19:05:21 浏览: 243
传送门机制的集合。这个存储库可能偶尔会更新.zip
以下是Python3的代码实现:
```python
from queue import Queue
def bfs(start, end, graph):
q = Queue()
q.put(start)
visited = [False] * len(graph)
visited[start] = True
dist = [0] * len(graph)
while not q.empty():
u = q.get()
for v in graph[u]:
if not visited[v]:
visited[v] = True
dist[v] = dist[u] + 1
q.put(v)
if v == end:
return dist[v]
return -1
n = int(input())
graph = [[] for _ in range(n)]
for i in range(n):
doors = list(map(int, input().split(',')))
for door in doors:
graph[i].append(door - 1)
start, end = map(int, input().split(','))
print(bfs(start - 1, end - 1, graph))
```
对于每个房间,我们可以用一个列表来表示该房间可以到达的房间编号。例如对于样例输入,我们可以用以下代码构造图:
```python
graph = [
[3, 5], # 房间1可以到达房间4和房间6
[1, 2, 3, 7], # 房间2可以到达房间1、房间2、房间4和房间8
[2, 5, 8], # 房间3可以到达房间3、房间6和房间9
]
```
然后我们可以使用BFS来求解从房间A到达房间B最少需要经过几个传送门。具体来说,我们用一个队列来存储当前遍历到的房间编号,用一个visited数组来记录每个房间是否已经被遍历过,用一个dist数组来记录从起点到每个房间的最短距离。每次从队列中取出一个房间u,遍历它可以到达的所有房间v,如果房间v还没有被遍历过,则将它加入队列中,并更新visited和dist数组。当我们遍历到终点房间时,直接返回dist数组中该房间的值即可。如果最后无法到达终点房间,则返回-1。
阅读全文