P1135 奇怪的电梯

奇怪的电梯

题目背景

感谢 @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;
}