奇怪的电梯

题目描述

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

输入格式

共二行。

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

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

输出格式

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

样例 #1

样例输入 #1

5 1 5
3 3 1 2 5

样例输出 #1

3

提示

对于 100%100 \%100% 的数据，1≤N≤2001 \le N \le 2001≤N≤200，1≤A,B≤N1 \le A, B \le N1≤A,B≤N，0≤Ki≤N0 \le K_i \le N0≤Ki​≤N。

思路

搜到了就返回按按钮次数，搜不到返回-1。注意判断是否出界和重复访问。

AC代码

#include 
#include 
#include 
#define AUTHOR "HEX9CF"
using namespace std;const int maxn = 1005;int k[1005];
int n, a, b;
queue q;
bool vis[maxn];
int cnt[maxn];int bfs(int a, int b){memset(vis, 0, sizeof(vis));memset(cnt, 0, sizeof(cnt));vis[a] = true;cnt[a] = 0;q.push(a);while(!q.empty()){int f = q.front();q.pop();// cout << f << endl;if(f == b){return cnt[b];}int t = f + k[f];if (t <= n && !vis[t]){cnt[t] = cnt[f] + 1;vis[t] = true;q.push(t);}t = f - k[f];if (t > 0 && !vis[t]){cnt[t] = cnt[f] + 1;vis[t] = true;q.push(t);}}return -1;
}int main() {cin >> n >> a >> b;for(int i = 1; i <= n; i++){cin >> k[i];}cout << bfs(a, b) << endl;return 0;
}