P1135 奇怪的电梯

CodeWizard～

于 2024-07-30 16:43:31 发布

阅读量363

点赞数 18

文章标签：算法深度优先图论数据结构 c++

本文链接：https://blog.csdn.net/2301_80662593/article/details/140609960

版权

奇怪的电梯

题目背景

感谢 @yummy 提供的一些数据。

题目描述

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

输入格式

共二行。

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

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

输出格式

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

样例 #1

样例输入 #1

5 1 5
3 3 1 2 5

样例输出 #1

提示

对于 $\%$ 的数据， $\le N \le 200$ ， $\le A, B \le N$ ， $\le K_i \le N$ 。

本题共 $16$ 个测试点，前 $15$ 个每个测试点 $6$ 分，最后一个测试点 $10$ 分。

没完全剪枝的code：

#include<iostream>
#include<algorithm>
using namespace std;
int n;//有n层楼
int A;//从第A层开始
int B;//目标是到第B层
int q[100010];//每一层可以走到的地方
int res=1e9;
void dfs(int x,int cnt){//x是当前所在的楼层，cnt是按了cnt次按钮
    if(cnt>=res) return ;
    if(x==B){
        res=min(res,cnt);
        return ;
    }
    if(x+q[x]<=n){
        dfs(x+q[x],cnt+1);
    }
    if(x-q[x]>0){
        dfs(x-q[x],cnt+1);
    }
}
int main(){
    cin>>n>>A>>B;
    for(int i=1;i<=n;i++) cin>>q[i];
    dfs(A,0);
    if(res==1e9){
        cout<<-1;
        return 0;
    }
    cout<<res;
    return 0;
}

剪枝后但还会tle六个数据的code：

由于每层楼走一次和每层楼走好多次然后绕回去的方案要好，所以可以实现像组合型枚举一样的方案，每个楼层只走一次，并设置一个数组来标记它。

#include<iostream>
#include<algorithm>
using namespace std;
int n;//有n层楼
int A;//从第A层开始
int B;//目标是到第B层
int q[100010];//每一层可以走到的地方
int res=1e9;
bool st[100010];
void dfs(int x,int cnt){//x是当前所在的楼层，cnt是按了cnt次按钮
    if(cnt>=res) return ;
    if(x==B){
        res=min(res,cnt);
        return ;
    }
    st[x]=true;
    //上
    if(x+q[x]<=n&&!st[x+q[x]]){
        st[x+q[x]]=true;
        dfs(x+q[x],cnt+1);
        st[x+q[x]]=false;
    }
    //下
    if(x-q[x]>0&&!st[x-q[x]]){
        st[x-q[x]]=true;
        dfs(x-q[x],cnt+1);
        st[x-q[x]]=false;
    }
}
int main(){
    cin>>n>>A>>B;
    for(int i=1;i<=n;i++) cin>>q[i];
    dfs(A,0);
    if(res==1e9){
        cout<<-1;
        return 0;
    }
    cout<<res;
    return 0;
}

剪枝正解的code：

本题可以定义一个ans[N]数组来存储每次到s层时按了几次按钮，当按按钮次数这次比上次还多时，之后的情况就可以pass掉了。(将ans[N]数组初始化为无穷，如果是无穷则代表电梯到达不了，返回-1，如果不是则返回ans[B])

#include<iostream>
#include<cstring>//memset函数
using namespace std;
const int N = 205;
int n;
int ans[N];
int A;
int B;
int dis[N]={0};
void dfs(int s,int t)
{
    ans[s]=t;
    if(s+dis[s]<=n&&t+1<ans[s+dis[s]]) dfs(s+dis[s],t+1);
    if(s-dis[s]>0&&t+1<ans[s-dis[s]]) dfs(s-dis[s],t+1);
}
int main()
{
    memset(ans,0x3f,sizeof(ans));
    cin>>n>>A>>B;
    for(int i=1;i<=n;i++) cin>>dis[i];
    dfs(A,0);
    if(ans[B]!=0x3f3f3f3f) cout<<ans[B];
    else cout<<-1;
    return 0;
}