奇怪的电梯(信息学奥赛一本通 - T1360)

题目描述
大楼的每一层楼都可以停电梯,而且第 i 层楼上有一个数字 Ki

电梯只有四个按钮:开,关,上,下,上下的层数等于当前楼层上的那个数字。

当然,如果不能满足要求,相应的按钮就会失灵。

例如:3 3 1 2 5 代表了 Ki(K1 = 3, K2 = 3, ……),从一楼开始。

在一楼,按 “上” 可以到 4 楼,按 “下” 是不起作用的,因为没有 −2 楼。

那么,从 A 楼到 B 楼至少要按几次按钮呢?

输入格式
第一行为三个用空格隔开的正整数,表示 N, A, B;
第二行为 N 个用空格隔开的正整数,表示 Ki

输出格式
一行,即最少按键次数,若无法到达,则输出 −1

输入样例
5 1 5
3 3 1 2 5

输出样例
3

数据范围
1 ≤ N ≤ 200
1 ≤ A, B ≤ N


题解
BFS:

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;

const int N = 210;

int n, a, b;
int w[N], dist[N];

int bfs()
{
	memset(dist, -1, sizeof dist);
	queue<int> q;
	q.push(a); 
	dist[a] = 0;
	
	while(q.size())
	{
		int t = q.front();
		q.pop();
		
		if(t == b) return dist[b];
		
		if(t + w[t] <= N && dist[t + w[t]] == -1)
		{
			q.push(t + w[t]);
			dist[t + w[t]] = dist[t] + 1;
		}
		
		if(t - w[t] >= 1 && dist[t - w[t]] == -1)
		{
			q.push(t - w[t]);
			dist[t - w[t]] = dist[t] + 1;
		}
	}
	
	return -1;
}

int main()
{
	cin >> n >> a >> b;
	
	for (int i = 1; i <= n; i ++) cin >> w[i];
	
	cout << bfs() << endl;
	return 0;
}
上一篇:406. 根据身高重建队列(贪心+简洁代码的线段树优化)


下一篇:KI子线段树 / AKEE SegmentTree