洛谷P1135 奇怪的电梯

题目描述

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

输入输出样例

输入 #1复制

5 1 5
3 3 1 2 5

输出 #1复制

3

思路：

用ans保存最小答案，并先给其赋值无穷大。用数组p判断是否来过下一层数，没有则进行操作。最后若ans不是无穷大，就进行输出。

代码如下：

#include <bits/stdc++.h>
using namespace std;
int n, a, b, ans = 0x7ffffff,k[205];
bool p[205];
void fun(int x, int sum)
{
	if (x == b) ans = min(ans, sum);
	if (sum > ans) return;
	p[x] = 1;
	if (x + k[x] <= n)
		if(p[x + k[x]] != 1) fun(x + k[x], sum + 1);
	if (x - k[x] >= 1 )
		if(p[x - k[x]] != 1) fun(x - k[x], sum + 1);
	p[x] = 0;
}
int main()
{
	cin >> n >> a >> b;
	p[a] = 1;
	for (int i = 1; i <= n; i++) cin >> k[i];
	fun(a,0);
	if (ans != 0x7ffffff) cout << ans;
	else cout << -1;
	return 0;
}

计算机202 张