洛谷1135 奇怪的电梯

题目描述

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

输入样例#1： 

5 1 5
3 3 1 2 5

输出样例#1： 

3

看到很多大佬都用最短路甚至DP来写这道题，本人不会只好用基本的宽搜了=-=

#include<iostream>
#include<queue>
using namespace std;
int n, st, en;
int vis[1000];
struct box
{
	int num;
	int k;
}tmp1,tmp2;    
int a[1000];
int main()
{
	cin >> n >> st >> en;
	queue<box> q;
	int flag = 0;
	for (int i = 1; i <= n; i++)
		cin >> a[i];
	tmp1.num=st,tmp1.k=a[st]; vis[st] = 0;            //将出发点压入队列并标记
	q.push(tmp1);
	while (!q.empty())
	{
		tmp1 = q.front(); q.pop();
		if (tmp1.num == en)                       //如果可以达到目标则标记一下
		{
			flag = 1;
			break;
		}
		tmp2.num = tmp1.num + tmp1.k;             
		if (tmp2.num <= n&&vis[tmp2.num]==0)      //第一种情况向上走，如果符合要求则压入队列并标记
		{
			tmp2.k =a[tmp2.num];
			q.push(tmp2);
			vis[tmp2.num] = vis[tmp1.num]+1;
		}
		tmp2.num = tmp1.num - tmp1.k;
		if (tmp2.num >= 1 && vis[tmp2.num] == 0)  //第二种情况向下走，如果符合要求则压入队列并标记
		{
			tmp2.k = a[tmp2.num];
			q.push(tmp2);
			vis[tmp2.num] = vis[tmp1.num]+1;      //注意无论向下走还是向上走都是属于“一步”的范围，标记数是相同的
		}
	}
	if (flag)
		cout << vis[en] << endl;
	else
		cout << "-1" << endl;
	return 0;
}