洛谷 P1135 奇怪的电梯
题目描述
呵呵,有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第i层楼(1<=i<=N)上有一个数字Ki(0<=Ki<=N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如:3 3 1 2 5代表了Ki(K1=3,K2=3,……),从一楼开始。在一楼,按“上”可以到4楼,按“下”是不起作用的,因为没有-2楼。那么,从A楼到B楼至少要按几次按钮呢?
分析
可用DFS做,稍微剪枝就行
#include <iostream>
#include <cstdio>
using namespace std;
int n,a[201],b[201],i,m,a1,b1,k;
void back(int high)
{
if (k>=m)
return;
if (high==b1)
{
m=k;
return;
}
b[high]=1;
k++;
if (high+a[high]<=n&b[high+a[high]]==0)
back(high+a[high]);
if (high-a[high]>=1&b[high-a[high]]==0)
back(high-a[high]);
k--;
b[high]=0;
}
int main()
{
m=10000;
scanf("%d%d%d",&n,&a1,&b1);
for (i=1;i<=n;i++)
scanf("%d",&a[i]);
back(a1);
if (m==10000)
printf("%d\n",-1);
else
printf("%d\n",m);
}