GYM 101147 E. Jumping(最短路径转化+spfa)

 

 有 n 个点,每个点 i 都可以传送到 i+x,或 i-x 的位置(每个点必须满足在区间 [1,n] 上),求每个点 i 到达 n 的最小传送次数

将 i -> i-x 转化为 起点为 i ,终点为 i-x,边权为 1 的路径,借助最短路径算法求解 

题目要求,从 i 到达 n 的最小路径,此时过程以 n 为起点返回 i,所以起点与终点与题意相反

const int N=2e5+5;
 
    int n,m,t;
    int i,j,k;
    int head[N],all=0;
    int d[N]; //点 i 到起点的距离
    bool vis[N];

struct Edge
{
    int to,next;
    int w;
}G[N];
void add(int u,int v,int w)
{
    G[all].w=w;
    G[all].to=v;
    G[all].next=head[u];
    head[u]=all++;
}
void SPFA(int s)
{
    for(int i=0;i<N;i++) vis[i]=0,d[i]=inf;
    d[s]=0; vis[s]=1;
    queue<int> q;
    q.push(s);
    while(!q.empty()){
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(int i=head[u];i!=-1;i=G[i].next){
            int v=G[i].to;
            int w=G[i].w;
            if(d[v]>d[u]+w){
                d[v]=d[u]+w;
                if(!vis[v]){
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
    }
}
int main()
{
    freopen("jumping.in","r",stdin);
    //IOS;
    rush(){
        sd(n);
        ms(head,-1);
        all=0;
        for(i=1;i<=n;i++){
            int x;
            sd(x);
            if(i-x>=1) add(i,i-x,1);
            if(i+x<=n) add(i,i+x,1);
        }
        SPFA(n);
        for(i=1;i<=n;i++){
            if(d[i]==inf) puts("-1");
            else pd(d[i]);
        }
    }
    //PAUSE;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/C_Dreamy/article/details/107864115