洛谷 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 楼至少要按几次按钮呢？

输入格式

共二行。

第一行为三个用空格隔开的正整数，表示N,A,B（1≤N≤200，1≤A,B≤N）。

第二行为N 个用空格隔开的非负整数，表示 Ki​。

输出格式

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

输入输出样例

输入 #1 复制

5 1 5

3 3 1 2 5

输出 #1 复制

3

说明/提示

对于 100%的数据，1≤N≤200，1≤A,B≤N，0≤Ki​≤N。

本题共 16 个测试点，前 15 个每个测试点 6分，最后一个测试点 10 分。
原题链接：https://www.luogu.com.cn/problem/P1135
这道题涉及的内容是搜索，我们可以使用bfs和dfs来解决这道问题

算法：BFS

时间复杂度：O(n)

你每一个点最多走一次，所以最差的情况就是每一个点都访问一次，才到达目标楼层，因此算法复杂度为O(n)，bfs可以确保我们第一个访问到的符合条件的点就是最优解

思路：

用一个bool 变量f 来表示能否从A走到B 

1.从A层开始，每次走一步
设当前楼层为now ,方向分为向上和向下，上（now+stair[now]）和 下（now-stair[now]）两个方向。

2.如果合法并且没有走过，就放入队列中，并将这一层标记已经访问过了
3.当队列为空时，还没有到达B，那么将无法到达B，就说明已经遍历所有可以走到的楼层，无解，输出-1.
4.注意：这种 bfs模板 做法在一开始的时候，A和B可能相等，这时候答案是0步，需要特判

代码实现

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> PII;
bool f=false; // 布尔变量f来表示能否从A到达B
const int N=205;
int n,A,B;
int k[N];
bool vis[N];
void bfs()
{
    queue<PII> q;
    q.push({A,0});
    if(A==B)
    {
        printf("0");
        f=true;
        return;             // 记得结束搜索
    }
    while(!q.empty())
    {
        PII t=q.front();
        q.pop();
        int now=t.first; int step=t.second; // now表示目前所在的楼层 step 表示所用的步数
        // 向上走
        int up=now+k[now];
        if(up>=1 && up<=n && !vis[up])
        {
            if(up==B)
            {
                printf("%d",step+1);
                f=true;
                return;     // 记得结束搜索
            }
            vis[up]=true;

            q.push({up,step+1});  // 继续扩展up所在这一层
        }
        
        //向下走
        int down=now-k[now];
        if(down>=1 && down<=n && !vis[down])
        {
            if(down==B)
            {
                printf("%d",step+1);
                f=true;
                return;     // 记得结束搜索
            }
            vis[down]=true;
            q.push({down,step+1});// 继续扩展down所在这一层
        }

    }
}
int main()
{
    cin>>n>>A>>B;
    for(int i=1;i<=n;i++) cin>>k[i];
    bfs();
    if(f==false) printf("-1");
    return 0;
}
//bfs的模板一般是通过队列来实现的

算法：DFS

 

改错题

我本来按照自己的理解写了dfs的代码，如下,发现TLE了，无奖改错，能帮我看看哪里出问题了吗？

#include<bits/stdc++.h>
using namespace std;
const int N=205;
int INF=0x3f3f3f3f; // 表示无穷大
int ans=INF; // 因为要找最小按键次数，因此设置初值为无穷大
int k[N];
bool vis[N];
int n,A,B;
void dfs(int now,int step)  //now 表示当前所在的楼层 step 表示已经走的步数
{
    if(now<1 || now>n) return;
    if(step>=ans) return;
    if(now==B)
    {
        ans=min(ans,step);
        return;
    }
    //向上搜索
    int up=now+k[now];
    if(up>=1 && up<=n && !vis[up])
    {
        vis[up]=true;
        dfs(up,step+1);
        vis[up]=false;
    }
    //向下搜索
    int down=now-k[now];
    if(down>=1 && down<=n && !vis[down])
    {
        vis[down]=true;
        dfs(down,step+1);
        vis[down]=false;
    }

}
int main()
{
    cin>>n>>A>>B;
    for(int i=1;i<=n;i++) cin>>k[i];
    dfs(A,0);
    if(ans==INF) printf("-1");
    else cout<<ans;
    return 0;
}

为什么会超时呢，是我剪枝剪的不够吗？时间复杂度我也拿捏的不是很准，有大佬帮忙看看哪里出问题了吗？

正确思路

讲一下正确的dfs做法，正确的剪枝，应当先判断可不可以更新，如果可以更新，那么再进行深层的搜索

代码实现

#include<bits/stdc++.h>
using namespace std;
const int N=205;
int n,A,B;
int dis[N];
int k[N];
void dfs(int now,int step)
{
    dis[now]=step;  // 因为我们是先判断可不可以更新在调用DFS,所以在这里一定可以更新
    // 向上搜索
    int up=now+k[now];
    if(up<=n && step+1<dis[up] )
    {
        dfs(up,step+1);
    }
    // 向下搜索
    int down=now-k[now];
    if(down>=1 && step+1<dis[down])
    {
        dfs(down,step+1);
    }
}
int main()
{
    memset(dis,0x3f3f,sizeof(dis)); // 初始化距离为无穷大
    cin>>n>>A>>B;
    for(int i=1;i<=n;i++) cin>>k[i];
    dfs(A,0);
    if(dis[B]==0x3f3f3f3f) printf("-1");
    else printf("%d",dis[B]);
    return 0;
}

这第二个代码太巧秒了