题目描述
呵呵,有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第 i 层楼(1≤1≤i≤N)上有一个数字 Ki(0≤0≤Ki≤N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如: 3,3,1,2,53,3,1,2,5 代表了 Ki(1=3K1=3,2=3K2=3,……),从 11 楼开始。在 11 楼,按“上”可以到 44 楼,按“下”是不起作用的,因为没有 −2−2 楼。那么,从 A 楼到 B 楼至少要按几次按钮呢?
输入格式
共二行。
第一行为三个用空格隔开的正整数,表示 N,A,B(1≤≤2001≤N≤200,1,≤1≤A,B≤N)。
第二行为 N 个用空格隔开的非负整数,表示 Ki。
输出格式
一行,即最少按键次数,若无法到达,则输出 -1
。
输入输出样例
输入 #1复制
5 1 5 3 3 1 2 5
输出 #1复制
3
说明/提示
对于 100%100% 的数据,1≤2001≤N≤200,1≤,≤1≤A,B≤N,0≤0≤Ki≤N。
本题共 1616 个测试点,前 1515 个每个测试点 66 分,最后一个测试点 1010 分。
答案:
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=300;
int n,a,b;
int q[N];
int res=1e9;
bool st[N];
void dfs(int x,int sum)
{
if(x>n) return;
if(sum>=res) return;
if(x<0 || x>n) return;
if(x==b)
{
res=min(res,sum);
return;
}
st[x]=true;
//上
if(x+q[x]<=n && !st[x+q[x]])
{
st[x+q[x]]=true;
dfs(x+q[x],sum+1);
st[x+q[x]]=false;
}
//下
if(x-q[x]>0 && !st[x-q[x]])
{
st[x-q[x]]=true;
dfs(x-q[x],sum+1);
st[x-q[x]]=false;
}
}
int main()
{
scanf("%d %d %d",&n,&a,&b);
for(int i=1;i<=n;i++)
{
scanf("%d",&q[i]);
}
dfs(a,0);
if(res==1e9)
printf("-1\n");
else printf("%d\n",res);
return 0;
}