题目描述
大楼的每一层楼都可以停电梯,而且第i层楼(1<=i<=N)上有一个数字Ki(0<=Ki<=N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如:3 3 1 2 5代表了Ki(K1=3,K2=3,……),从一楼开始。在一楼,按“上”可以到4楼,按“下”是不起作用的,因为没有-2楼。那么,从A楼到B楼至少要按几次按钮呢?
输入
输入文件共有二行,第一行为三个用空格隔开的正整数,表示N,A,B(1≤N≤200, 1≤A,B≤N),第二行为N个用空格隔开的正整数,表示Ki。
输出
输出文件仅一行,即最少按键次数,若无法到达,则输出-1。
样例输入
5 1 5
3 3 1 2 5
样例输出
3
解决这个问题我们可以先读入输入数据,并根据题目要求,计算任意两层楼之间的最短距离,即距离矩阵d[i][j]。
由于电梯只有上下两个按钮,所以任意两层楼之间的最短路程需要乘以2才是真正的最少按键次数。所以我们可以得到从A楼到B楼的最少按键次数为:
需要注意的是,当A楼或B楼上没有电梯时,无法到达目标楼层,此时应输出-1。
另外,这个问题在最坏情况下需要计算n的三次方次,时间复杂度较高,如果数据规模很大的话,可能需要考虑优化算法。
输入部分:
n, a, b = map(int, input().split())
k = list(map(int, input().split()))
初始化距离矩阵和访问列表:
d = [[float('inf')] * n for _ in range(n)]
vis = [False] * n
计算任意两层楼之间的最短距离:
for i in range(n):
vis[i] = True
for j in range(n):
if vis[j]:
d[i][j] = abs(i-j)
vis[i] = False
for k1 in range(n):
for k2 in range(n):
for k3 in range(n):
d[k2][k3] = min(d[k2][k3], d