洛谷 [P1135 奇怪的电梯] {搜索|BFS} 奋斗的珂珂~

题目描述

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

输入输出样例

输入
5 1 5
3 3 1 2 5
输出
3

完整代码

#include<bits/stdc++.h>
using namespace std;
//bfs求最少次数
int vis[250];//vis表示该楼层是否到达过 
int flor[250];

struct elevator{
	int id;//代表楼层号 
	int step;//代表走的几步 
}begin; 

int main()
{
	int N,A,B;
	scanf("%d %d %d",&N,&A,&B);
	for(int i=1;i<=N;i++)
	{
		scanf("%d",&flor[i]);
	}
	queue<elevator> p;
	begin.id=A;
	begin.step=0;
	p.push(begin);//初始入队 
	elevator e;
	while(!p.empty())
	{
		e=p.front();
		if(e.id==B) break;
		p.pop();
		if(e.id+flor[e.id]<=N&&!vis[e.id+flor[e.id]])//题目中说明可到达楼层在1~N之间，故可作为判定的条件 
		{
			vis[e.id+flor[e.id]]=1;
			p.push((elevator){e.id+flor[e.id],e.step+1}); //如果可行，就入队列 
		}
		if(e.id-flor[e.id]>=1&&!vis[e.id-flor[e.id]])
		{
			vis[e.id-flor[e.id]]=1;
			p.push((elevator){e.id-flor[e.id],e.step+1});
		}	
	}
	if(e.id==B) printf("%d",e.step);//判定是一位有解输出还是误无解循环终止导致输出 
	else printf("-1");
	return 0;
 }