球球大佬救救我这个菜鸡
查看原帖
球球大佬救救我这个菜鸡
477670
roseCoder楼主2021/4/17 18:10

代码写成这样所有测试点都超时了,蹲一个大佬帮我看一下。

#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;

const int N = 210;
int w[N];
int dist[N];
int n;
void bfs(int x)    //深度优先搜索
{
    memset(dist,0x3f3f3f,sizeof dist); //初始化为一个大数
    dist[x] = 0;                       //x位置距离为0
    queue<int> q;
    q.push(x);
    while(q.size()){
        auto t = q.front();
        q.pop();
        int a = t+w[t];                //a记录在t层上升的层数
        if (a>=1&&a<=n){               //判断是否符合
            q.push(a);
            dist[a] = min(dist[a],dist[t]+1);//a层的距离为之前的最小值和t的距离+1的最小值
        }
        a = t-w[t];                    //a换成t层下降的层数
        if (a>=1&&a<=n){               //判断是否符合
            q.push(a);
            dist[a] = min(dist[a],dist[t]+1);
        }
    }
}
int main()
{
    cin>>n;
    int a, b;
    cin>>a>>b;
    for (int i=1;i<=n;i++) cin>>w[i];
    bfs(a);                  //深度优先搜索
    if (a==b) cout<<0<<endl;
    else if(dist[b]==0x3f3f3f) cout<<-1<<endl;
    else cout<<dist[b]<<endl;

    return 0;
}
2021/4/17 18:10
加载中...