洛谷 P1135 奇怪的电梯

传送门：P1135 奇怪的电梯

题目描述

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

输入格式

共二行。

第一行为三个用空格隔开的正整数，表示 。

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

输出格式

一行，即最少按键次数，若无法到达，则输出 -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 分。

----------------------------------------------------------分割线--------------------------------------------------------------

知识点 

宽度优先搜索（BFS）

解题思路

以A为起始楼层，开始时入队A。在后续按电梯的过程中，由于分为上下楼，每一层的数字都不一样，故将设置i初始化为-1，i=-1时表示下楼，i=1时表示上楼，从而达到上下楼的目的。最后利用ans[]数组储存结果，如果目标楼层B未达到，则ans[B]即为初始化时的-1，并输出-1。如果达到B楼层，则输出答案。

AC代码

#include<bits/stdc++.h>
using namespace std;
int N, A, B,k[205];//N为楼层，A为起始楼层，B为目标楼层,k[]为按钮数字
int ans[205];
bool vis[205];
void bfs() {
	queue<int>q;
	q.push(A);
	vis[A] = true;//标记A层来过
	ans[A] = 0;
	while (!q.empty()) {
		int floor = q.front();
		q.pop();
		for (int i = -1; i <= 1; i+=2) {//i = -1 表示下楼，i = 1 表示上楼
			int next_floor = floor + k[i]*i;
			if (next_floor > N || next_floor < 0 || vis[i])
				continue;//如果楼层不合法，或已来过，则返回上一个状态
			vis[next_floor] = true;
			ans[next_floor]=ans[floor]+1;
			q.push(next_floor);
		}
	}
}
int main() {
	cin >> N >> A >> B;
	memset(ans, -1, sizeof(ans));
	for (int i = 1; i <= N; i++) {
		cin >> k[i];
	}
	bfs();
	if (ans[B] == -1)//如果没达到，B楼层的步数即为初始化时赋值的-1
		cout << -1 << endl;
	else
		cout << ans[B];
}