原题出处:洛谷题单
题目:奇怪的电梯
呵呵,有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第ii层楼(1 \le i \le N)(1≤i≤N)上有一个数字K_i(0 \le K_i \le N)K i (0≤K i ≤N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如:3, 3 ,1 ,2 ,53,3,1,2,5代表了K_i(K_1=3,K_2=3,…)K
i (K 1=3,K 2 =3,…),从11楼开始。在11楼,按“上”可以到44楼,按“下”是不起作用的,因为没有-2−2楼。那么,从AA楼到BB楼至少要按几次按钮呢?
输入格式
共二行。
第一行为33个用空格隔开的正整数,表示N,A,B(1≤N≤200, 1≤A,B≤N)N,A,B(1≤N≤200,1≤A,B≤N)。
第二行为N个用空格隔开的非负整数,表示Ki。
输出格式
一行,即最少按键次数,若无法到达,则输出−1。
输入输出样例
输入 #1
5 1 5
3 3 1 2 5
输出 #1
3
DFS解法:
1、首先得把题意看懂了,一共n层楼,要从A曾到达B层,因为电梯的特殊规则,所以,在尝试的时候,可能会遇到无法到达或者多种方式到达的情况,如果把特殊情况考虑到位起码可以得到20分:
第一种:A=B时,输出是0 (10分)
第二种:所有楼层都试了就是没有一条能到达B,那么cnt就始终等于它的初始值,这里你要赋给它一个很大的初始值。(10分)
2、用搜索的思想来解决,就比较简单了直接套用dfs的搜索框架就行了
(我写完到这里的时候,有两个测试点超时了)
3、解决超时问题就得想想怎么样来简化它,第一,我要找最小值,所以如果在搜索过程中出现的案件次数已经比我之前存好的cnt大了,那么这个时候就没必要沿着这条路走了可以直接return,到这里ok了,可以AC。
#include<bits/stdc++.h>
using namespace std;
int n,A,B,cnt=100001;
int k[210];
bool b[210]={0};//标记楼层是否已经走过
int d[2]={1,-1};//上楼加,下楼减
void dfs(int x,int s)
{
if(x==B) cnt=min(s,cnt);//如果已经到达了
if(s>cnt) return;//如果s大于cnt就不用再继续了,我们要找的是最小值
//下面是典型的dfs
for(int i=0;i<2;i++){
int xx=x+d[i]*k[x];
if(xx>=1 && xx<= n && b[xx]){
s++;
b[x]=false;
dfs(xx,s);
b[x]=true;//回溯
s--;//回溯
}
}
}
int main(){
cin>>n>>A>>B;
memset(b,true,sizeof(b));
for(int i=1;i<=n;i++) cin>>k[i];
b[A]=false;//标记出发楼层 ,这里是容易忽略的哦
if(A==B) cnt=0;//注意:如果当前楼层和要到达的楼层是一样的cnt=0
else dfs(A,0);
if(cnt==100001) cout<<-1;//注意:如果cnt这这个值的话说明没有找到一条路径,这里因为数值范围小,所以100001肯定够了
else cout<<cnt;
return 0;
}
BFS解法见下一篇博文:这道题的BFS解法,简单易懂