洛谷 P1135 奇怪的电梯 （dfs）

题目描述

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

输入输出格式

输入格式：

输入文件共有二行，第一行为三个用空格隔开的正整数，表示N,A,B(1≤N≤200, 1≤A,B≤N)，第二行为N个用空格隔开的正整数，表示Ki。

输出格式：

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

输入输出样例

输入样例#1：

LIFT.IN
5 1 5
3 3 1 2 5

输出样例#1：

LIFT.OUT
3

#include<cstdio>
#include<iostream>

using namespace std;

int a,b,n,k,qwq=0x7fffffff;
int up[100001],down[100001];
int qmq[100001];

void dfs(int s,int c)
{
    if(s==b) 
    {
        if(c<qwq) qwq=c;
        return;
    }
    if(c>qwq)
    return;
    if(!qmq[s]&&up[s]!=0&&up[s]!=s)
    {
        qmq[s]=1;
        dfs(up[s],c+1);
        qmq[s]=0;//回溯，减少不必要的麻烦 
    }
    if(!qmq[s]&&down[s]!=0&&down[s]!=s)
    {
        qmq[s]=1;
        dfs(down[s],c+1);
        qmq[s]=0;
    }
    return;
}

int main()
{
    scanf("%d %d %d",&n,&a,&b);
    if(a<1||b>n)//剪枝 
    {
        printf("-1\n");
        return 0;
    }//必须在范围之内
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&k);
        if(i+k<=n) up[i]=i+k;
        if(i>k) down[i]=i-k;
    }
    dfs(a,0);
    if(qwq!=0x7fffffff) printf("%d",qwq);
    else printf("-1\n"); 
    return 0;
}