奇怪的电梯
题目描述
呵呵,有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第ii层楼(1 \le i \le N)(1≤i≤N)上有一个数字K_i(0 \le K_i \le N)Ki(0≤Ki≤N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如:3, 3 ,1 ,2 ,53,3,1,2,5代表了K_i(K_1=3,K_2=3,…)Ki(K1=3,K2=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)。
第二行为NN个用空格隔开的非负整数,表示K_iKi。
输出格式:
一行,即最少按键次数,若无法到达,则输出-1−1。
输入输出样例
输入样例#1: 复制
5 1 5 3 3 1 2 5
输出样例#1: 复制
3
思路:
记忆化搜索,边走边标记,加上剪枝。
参考代码:
#include<cstdio> using namespace std; int n,a,b; int k[239],f[239],ans=0x7fffffff; void dfs(int num,int test){ if(test>=ans)return; if(num<1||num>n+1||f[num])return; if(num==b){ if(test<ans)ans=test; return; } f[num]=1; if(num-k[num]>0)dfs(num-k[num],test+1); if(num+k[num]<n+1)dfs(num+k[num],test+1); f[num]=0; } int main(){ scanf("%d%d%d",&n,&a,&b); for(int i=1;i<=n;i++) scanf("%d",&k[i]); dfs(a,0); printf("%d",ans!=0x7fffffff?ans:-1); }