CSP/信奥赛C++刷题训练：经典广搜例题（2）：洛谷P1135 ：奇怪的电梯
题目背景
感谢 @yummy 提供的一些数据。
题目描述
呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第 i i i 层楼（ 1 ≤ i ≤ N 1 \le i \le N 1≤i≤N）上有一个数字 K i K_i Ki（ 0 ≤ K i ≤ N 0 \le K_i \le N 0≤Ki≤N）。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如： 3 , 3 , 1 , 2 , 5 3, 3, 1, 2, 5 3,3,1,2,5 代表了 K i K_i Ki（ K 1 = 3 K_1=3 K1=3， K 2 = 3 K_2=3 K2=3，……），从 1 1 1 楼开始。在 1 1 1 楼，按“上”可以到 4 4 4 楼，按“下”是不起作用的，因为没有 − 2 -2 −2 楼。那么，从 A A A 楼到 B B B 楼至少要按几次按钮呢？
输入格式
共二行。
第一行为三个用空格隔开的正整数，表示 N , A , B N, A, B N,A,B（ 1 ≤ N ≤ 200 1 \le N \le 200 1≤N≤200， 1 ≤ A , B ≤ N 1 \le A, B \le N 1≤A,B≤N）。
第二行为 N N N 个用空格隔开的非负整数，表示 K i K_i Ki。
输出格式
一行，即最少按键次数，若无法到达，则输出 -1。
样例 #1
样例输入 #1
5 1 5
3 3 1 2 5
样例输出 #1
3
提示
对于 100 % 100 \% 100% 的数据， 1 ≤ N ≤ 200 1 \le N \le 200 1≤N≤200， 1 ≤ A , B ≤ N 1 \le A, B \le N 1≤A,B≤N， 0 ≤ K i ≤ N 0 \le K_i \le N 0≤Ki≤N。
本题共 16 16 16 个测试点，前 15 15 15 个每个测试点 6 6 6 分，最后一个测试点 10 10 10 分。
使用广搜解题
#include<bits/stdc++.h>
using namespace std;
//广搜思路：用队列维护节点和步数
int n,a,b,k[210];
bool f=1,u[210];//u是标记数组
int q[410][2];//q数组模拟队列
int main(){
cin>>n>>a>>b;
for(int i=1;i<=n;i++) cin>>k[i];
//广搜
int head=1,tail=1;//头尾指针
q[1][0]=a;//起点入队
q[1][1]=0;//起点步数初始化为0
while(head<=tail){
int x=q[head][0];//取当前结点
int y=q[head][1];
if(x==b){
cout<<y;//如果到达b，输出答案
f=0;//标记有答案
return 0;//找到答案就可以结束程序了
}
//上
int nx=x+k[x];
int ny=y+1;
if(nx<=n && !u[nx]){
tail++;
q[tail][0]=nx;
q[tail][1]=ny;
u[nx]=1;//标记
}
//下
nx=x-k[x];
ny=y+1;
if(nx>=1 && !u[nx]){
tail++;
q[tail][0]=nx;
q[tail][1]=ny;
u[nx]=1;//标记
}
head++;//队首出队
}
if(f) cout<<-1;
return 0;
}
文末彩蛋：