奇怪的电梯
题目背景
感谢 @yummy 提供的一些数据。
题目描述
呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第 \(i\) 层楼（\(1 \le i \le N\)）上有一个数字 \(K_i\)（\(0 \le K_i \le N\)）。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如： \(3, 3, 1, 2, 5\) 代表了 \(K_i\)（\(K_1=3\)，\(K_2=3\)，……），从 \(1\) 楼开始。在 \(1\) 楼，按“上”可以到 \(4\) 楼，按“下”是不起作用的，因为没有 \(-2\) 楼。那么，从 \(A\) 楼到 \(B\) 楼至少要按几次按钮呢？
输入格式
共二行。
第一行为三个用空格隔开的正整数，表示 \(N, A, B\)（\(1 \le N \le 200\)，\(1 \le A, B \le N\)）。
第二行为 \(N\) 个用空格隔开的非负整数，表示 \(K_i\)。
输出格式
一行，即最少按键次数，若无法到达，则输出 -1。
样例 #1
样例输入 #1
5 1 5
3 3 1 2 5
样例输出 #1
3
提示
比较简单的一道BFS题
using namespace std;
int p[1000000];
int vis[10000];
int n,a,b;
struct node{
int t,id;
node(){}
node(int a,int b){
t=b;
id=a;
}
};
bool BFS(){
queue<node>q;
q.push(node(a,0));
vis[a]=1;
while(q.size()){
node x=q.front();
q.pop();
int h=x.id;
if(h==b){
cout<<x.t;
return true;
}
int nx1=h+p[h];
int nx2=h-p[h];
if(nx1<=n&&!vis[nx1]){
q.push(node(nx1,x.t+1));
vis[nx1]=1;
}
if(nx2>0&&!vis[nx2]){
q.push(node(nx2,x.t+1));
vis[nx2]=1;
}
}
return false;
}
int main(){
cin>>n>>a>>b;
for(int i=1;i<=n;i++)cin>>p[i];
if(!BFS()){
cout<<-1;
}
return 0;
}