题目大意:
高度为n的每层电梯都可以上升ki层或者下降ki层,问你是否能从a层坐电梯到b层,如果可以,问你最少要在几个楼层停留,如果不存在,输出-1
解题思路:
bfs+优先级队列!
#include<stdio.h>
#include<queue>
#include<string.h>
#define N 210
using namespace std;
struct node{
int cur;
int sum;
bool operator <(const node &a) const{
return sum>a.sum;
}
};
int k[N],a,b,n,ans;
bool visit[N];
int bfs(){
node temp={a,0};
int i,sum;
priority_queue<node> q;
q.push(temp);
visit[a]=true;
while(!q.empty()){
temp=q.top();
q.pop();
i=temp.cur;sum=temp.sum;
if(i==b)
return sum;
if(i+k[i]<=n&&!visit[i+k[i]]){
temp.cur=i+k[i];temp.sum=sum+1;
q.push(temp);
visit[i+k[i]]=true;
}
if(i-k[i]>0&&!visit[i-k[i]]){
temp.cur=i-k[i];temp.sum=sum+1;
q.push(temp);
visit[i=k[i]]=true;
}
}
return -1;
}
int main()
{
int i;
while(scanf("%d",&n)&&n){
scanf("%d%d",&a,&b);
for(i=1;i<=n;i++)
scanf("%d",&k[i]);
memset(visit,0,sizeof(visit));
ans=bfs();
printf("%d\n",ans);
}
}