P1135 奇怪的电梯 c++

题目描述

呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第 i层楼（1≤i≤N）上有一个数字 Ki​（0 \le K_i \le N0≤Ki​≤N）。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如： 3, 3, 1, 2, 代表了 Ki​（K1​=3，K2​=3，……），从 1 楼开始。在 1 楼，按“上”可以到 4 楼，按“下”是不起作用的，因为没有 -2 楼。那么，从 A楼到 B 楼至少要按几次按钮呢？

输入格式

共二行。

第一行为三个用空格隔开的正整数，表示 N, A, B（1≤N≤200，1≤A,B≤N）。

第二行为 NN 个用空格隔开的非负整数，表示Ki​。

输出格式

一行，即最少按键次数，若无法到达，则输出 -1。

输入输出样例

输入 #1

5 1 5
3 3 1 2 5

输出 #1

3

说明/提示

对于 100% 的数据，1≤N≤200，1≤A,B≤N，0≤Ki​≤N。

本题共 16 个测试点，前 15个每个测试点 6 分，最后一个测试点 10 分。

#include<iostream>
#include<queue>
using namespace std;
int n,dir[201],v[201],A,B;
struct point{
	int x,step;
	point(int xx,int st){
		x=xx;
		step=st;
	}
};
int bfs(int x){
	queue<point> q;
	q.push(point(x,0));
	v[x]=1;
	while(!q.empty()){
		point p=q.front();
		int m=p.x,s=p.step;
		if(m==B) return s;
		if(m+dir[m]>0&&m+dir[m]<=n&&!v[m+dir[m]]){
			q.push(point(m+dir[m],s+1));
			v[m+dir[m]]=1;
		}
		if(m-dir[m]>0&&m-dir[m]<=n&&!v[m-dir[m]]){
			q.push(point(m-dir[m],s+1));
			v[m-dir[m]]=1;
		}
		q.pop();
	} 
	return -1;
}
int main(){
	cin>>n>>A>>B;
	for(int i=1;i<=n;i++) cin>>dir[i];
	cout<<bfs(A);
	return 0;
}