奇怪的电梯

题目描述

有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第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。

样例

input

5 1 5
3 3 1 2 5

output

3

 思路

考虑用 BFS 实现，用队列存储从起点开始按 t 次电梯后能到达的状态。

首先将起点入队（当然，起点和终点相等时直接输出 0 即可），接着重复循环，每次将队列中所有状态向前推一步。

• 如果到达了之前已经到达过的状态，那么没必要推下去，重复之前走过的路。
• 如果到达终点，直接输出当前一共向前推了几步。
• 如果到某个时刻队列为空，说明这一步没有可以到达的新的楼层了，而此时仍然没有到达目标楼层，所以直接输出 -1。
• N≤200，而每个楼层最多访问一次，所以不用担心超时。

代码

#include<bits/stdc++.h>
using namespace std;
int a[0xcff0102];
bool vis[0xcff0102];
int main(){
	ios::sync_with_stdio(false);
	int n,s,e;cin>>n>>s>>e;
	if(s==e){//初始楼层和目标楼层相同，不需要按电梯 
		cout<<0;
		return 0;
	} 
	for(int i=1;i<=n;i++)cin>>a[i];//输入
	queue<int>q;
	q.push(s);
	int t=0;vis[s]=1;
	while(++t){
		int x=q.size();
		if(x==0){//空了
			cout<<-1;
			return 0;
		}
		for(int i=0;i<x;i++){//将当前 q 中所有状态向前推一遍
			int p=q.front();
			q.pop();
			if(p-a[p]>0){//按下键时电梯不会跑到地底下 
				if(!vis[p-a[p]]){//这个状态没被访问过 
					if(p-a[p]==e){//到目标楼层啦 
						cout<<t;
						return 0;
					} 
					vis[p-a[p]]=1;
					q.push(p-a[p]);
				}
			}
			if(p+a[p]<=n){//按上键时电梯不会飞到天上 
				if(!vis[p+a[p]]){//这个状态没被访问过
					if(p+a[p]==e){//到目标楼层啦 
						cout<<t;
						return 0;
					} 
					vis[p+a[p]]=1;
					q.push(p+a[p]);
				}
			}
		}
	}
	return 0;
}