P1135 奇怪的电梯（BFS/DFS）

来源：https://www.luogu.org/problemnew/show/P1135

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

输入
共二行。

第一行为3个用空格隔开的正整数，表示N,A,B(1≤N≤200, 1≤A,B≤N) ,第二行为N个用空格隔开的非负整数，表示K i。

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

样例输入
5 1 5
3 3 1 2 5
样例输出
3
解法一：BFS
AC_code:

#include <bits/stdc++.h>
using namespace std;
int n,a,b;
int g[205];
bool vis[205];
int f[205],flag;
int BFS()
{
    queue<int>q;
    q.push(a);
    f[a] = 0;
    while(!q.empty())
    {
        int k = q.front();
        if(k == b)
        {
            return f[k];
        }
        //vis[k] = true;
        //vis单独放在这标记是不对的，Ac90%,应该是入队就标记
        q.pop();
        int x = k + g[k], y = k - g[k];
        if(x<=n && !vis[x])
        {
            vis[x] = true;
            q.push(x);
            f[x] = f[k]+1;
            
        }
        if(y>=1 && !vis[y])
        {
            vis[y] = true;
            q.push(y);
            f[y] = f[k]+1;
        }
        
    }
    return -1;
    
}
int main()
{
    cin>>n>>a>>b;
    memset(f,0,sizeof(f));
    memset(vis,false,sizeof(vis));
    for(int i = 1; i <= n; i++)
    {
        cin>>g[i];
    }
    int ans = BFS();
    cout<<ans<<endl;
    return 0;
}

解法二：
DFS:

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
vector<int>v[205];
bool vis[205];
int n,a,b,ans = INF ;
void DFS(int s,int cnt)
{
    if(cnt >= ans)
        return;
    if(s == b)
    {
        ans = cnt;
        return;
    }
    for(int i = 0; i < v[s].size(); i++)
    {
        if(!vis[v[s][i]])
        {
            vis[v[s][i]] = true;
            DFS(v[s][i],cnt+1);
            vis[v[s][i]] = false;
        }
    }
}
int main()
{
    cin>>n>>a>>b;
    int x;
    for(int i = 1; i <= n; i++)
    {
        cin>>x;
        int y = i + x, z = i - x;
        if(y<=n)
            v[i].push_back(y);
        if(z>=1)
            v[i].push_back(z);
    }
    DFS(a,0);
    if(ans == INF)
        cout<<-1<<endl;
    else
        cout<<ans<<endl;
    return 0;
}