洛谷P1135 奇怪的电梯bfs

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n, a, b;
int arr[201], ans[201], fw[201];//记录输入，答案，访问数组
queue<int>q;
int main()
{
	cin >> n >> a >> b;
	for (int i = 1; i <= n; i++)
	{
		cin >> arr[i];
	}
	memset(ans, -1, sizeof(ans));//初始化-1
	q.push(a);
	ans[a] = 0;//初始楼层进队且初始化答案为0
	while (!q.empty())
	{
		int lc = q.front();//取出首位
		q.pop();//弹出首位
		if (lc == b)break;//如果到了目的地退出循环
		for (int i = -1; i <= 1; i += 2)
		{
			int lc2 = lc + (i * arr[lc]);//模拟上下楼
			if (lc2 >= 1 && lc2 <= n && ans[lc2] == -1)
			{
				ans[lc2] = ans[lc] + 1;//新楼层是上一个楼层加一
				q.push(lc2);//新楼层入队
			}
		}
	}
	cout << ans[b];
}