Python编程问题之奇怪的电梯

题目描述
大楼的每一层楼都可以停电梯，而且第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