【题解】helloworldroom#2485. 奇怪的电梯

题目链接

#2485. 奇怪的电梯http://go.helloworldroom.com:50080/problem/2485

#2485.奇怪的电梯

题目描述

有一天桐桐做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第i层楼（1≤i≤N）上有一个数字Ki（0≤Ki≤N）。

电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层加上或减去那个数字。当然，如果不能满足要求，相应的按钮就会失灵。

例如：3 3 1 2 5代表了Ki（K1=3，K2=3，…），从一楼开始。在一楼，按“上”可以到4楼，按“下”是不起作用的，因为没有-2楼。

那么，从A楼到B楼至少要按几次按钮呢？

输入格式

第1行为三个正整数，表示N，A，B（1≤N≤200，1≤A，B≤N）；

第2行为N个正整数，表示Ki。

输出格式

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

样例数据

input

5 1 5
3 3 1 2 5

 output

3

题目分析

基于这道题数据量并不大，我们可以用宽搜BFS来解决这道题，再配合上队列queue容器。不断向队列压入子元素，弹出父元素，直到到达题目要求的层数或队列为空为止

题目代码

#include<bits/stdc++.h>
using namespace std;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int lcm(int a,int b){return a/(gcd(a,b))*b;}
int n,a,b;
int k[201];
bool visited[201];
struct node{
	int floor;
	int num;
};
queue <node> q;
node x,y;

int main(){
	cin>>n>>a>>b;
	for(int i=1;i<=n;i++)cin>>k[i];
	if(a==b){
		cout<<0<<endl;
		return 0;
	}
	x.floor=a;x.num=0;
	visited[a]=1;
	q.push(x);
	while(!q.empty()){
		x=q.front();
		y.floor=x.floor+k[x.floor];
		y.num=x.num+1;
		if(y.floor<=n&&!visited[y.floor]){
			q.push(y);
			visited[y.floor]=1;
			if(y.floor==b){
				cout<<y.num<<endl;
				return 0;
			}
		}
		y.floor=x.floor-k[x.floor];
		y.num=x.num+1;
		if(y.floor>0&&!visited[y.floor]){
			q.push(y);
			visited[y.floor]=1;
			if(y.floor==b){
				cout<<y.num<<endl;
				return 0;
			}
		}
		q.pop();
	}
	cout<<-1<<endl;
	return 0;
}