奇怪的电梯
呵呵,有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第i层楼(1≤i≤N)上有一个数字Ki(0≤Ki ≤N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如:3, 3 ,1 ,2 ,5代表了Ki(K1=3,K2=3,…),从1楼开始。在1楼,按“上”可以到4楼,按“下”是不起作用的,因为没有−2楼。那么,从A楼到B楼至少要按几次按钮呢?
输入格式:
共二行。
第一行为3个用空格隔开的正整数,表示N,A,B(1≤N≤200,1≤A,B≤N)。
第二行为N个用空格隔开的非负整数,表示Ki 。
输出格式:
一行,即最少按键次数,若无法到达,则输出−1。
样例.in
5 1 5
3 3 1 2 5
样例.out
3
代码
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
int n,a,b,kk[210];//kk为楼层按钮上的数
bool v[210];
struct node
{
int f,k;
};
queue<node> q;//定义队列
int main()
{
cin >>n>>a>>b;//a一层,b目标楼层
for(int i=1;i<=n;i++)//读入每层数字
cin>>kk[i];
//以下bfs
q.push((node){a,0});// 放入初始楼层,记录到这个楼层按了几次
while(!q.empty()) //当队列里非空时
{
node u=q.front();//从队列中取出队头数据
q.pop();
if(u.f==b)//判断楼层是否是目标楼层
{
cout<<u.k;//是,输出按了几次
return 0;
}
//否则看这层能去那几层,放入队列
int df=u.f+kk[u.f];//u.f为当前层,kk为当前层按钮数字
//向上去
if(df<=n/*是否超过最高层*/&&!v[df]/*楼层是否走过*/)
{
v[df]=true;
q.push((node){df,u.k+1/*到它的下一层,增加一步按按钮次数*/});
}
df=u.f-kk[u.f];//向下去
if(df>=1&&!v[df])
{
v[df]=true;
q.push((node){df,u.k+1});
}
}
cout<<-1;//如果while没有找到就返回-1
return 0;
}