P1135 奇怪的电梯

题目描述

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

输入输出格式

输入格式：

 

共二行。

第一行为3个用空格隔开的正整数，表示N,A,B(1≤N≤200, 1≤A,B≤N)。

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

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

5 1 5
3 3 1 2 5

输出样例#1： 复制

3

 

---

一开始的写法，数据是可以过的，但是递归太深会爆栈，MLE(MEMORY LIMIT EXCEEDED)

#include<iostream>
#include<algorithm>
#define inf 0x3f3f3f3f
using namespace std;
int n,a,b; //n表示有几层楼，a、b是目标楼层 
int f[205];//记录每一层的k值 
int ans=inf;

void dfs(int now,int step) //now是现在所在的楼层  step是按键次数 
{
	if(now==b)
	{
		ans=min(ans,step);
		return;
	}
	if(now+f[now]<=n) //可以上升 
		dfs(now+f[now],step+1);
	if(now-f[now]>=1)//可以下降
		dfs(now-f[now],step+1);	
}
//这样写  爆栈了 ，最后MLE 
int main()
{
	ios::sync_with_stdio(false);
	cin>>n>>a>>b;
	for(int i=1;i<=n;i++)
		cin>>f[i];		
	dfs(a,0);
	cout<<ans<<endl;
	return 0;
}

 

后来改了写法：

#include<iostream>
#include<algorithm>
#define inf 0x3f3f3f3f
using namespace std;
int n,a,b; //n表示有几层楼，a、b是目标楼层 
int f[205];//记录每一层的k值 
int ans=inf;
int vis[205];

void dfs(int now,int step) //now是现在所在的楼层  step是按键次数 
{
	if(now==b)
	{
		ans=min(ans,step);
		return;
	}
	if(step<=ans) //这就是一个剪枝
	{
		vis[now]=1;
		if(now+f[now]<=n && !vis[now+f[now]]) //可以上升，还要注意判断一下是否访问过(也是一种剪枝) 
			dfs(now+f[now],step+1);
		if(now-f[now]>=1 && !vis[now-f[now]])//可以下降
			dfs(now-f[now],step+1);	
		vis[now]=0;	
	}
}

int main()
{
	ios::sync_with_stdio(false);
	cin>>n>>a>>b;
	for(int i=1;i<=n;i++)
		cin>>f[i];		
	dfs(a,0);
	if(ans!=inf)
		cout<<ans<<endl;
	else  //别忘了判断输出-1 
		cout<<-1<<endl;
	return 0;
}