考试期间来电脑室,嚣不嚣张?

题目描述

如题,给出一个有向图,请输出从某一点出发到所有点的最短路径长度。

输入输出格式

输入格式:
第一行包含三个整数N、M、S,分别表示点的个数、有向边的个数、出发点的编号。

接下来M行每行包含三个整数Fi、Gi、Wi,分别表示第i条有向边的出发点、目标点和长度。

输出格式:
一行,包含N个用空格分隔的整数,其中第i个整数表示从点S出发到点i的最短路径长度(若S=i则最短路径长度为0,若从点S无法到达点i,则最短路径长度为2147483647)

说明

时空限制:1000ms,128M

数据规模:

对于20%的数据:N<=5,M<=15

对于40%的数据:N<=100,M<=10000

对于70%的数据:N<=1000,M<=100000

对于100%的数据:N<=10000,M<=500000

思路:

spfa。

对于这个算法,是非常高效的。
先用一个队列来维护活跃点(bfs),对于每次更改,都要做一次松弛操作。
O(km) k<5;

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
long long next[511111],u[511111],d[511111],v[511111],w[511111],state[511111],list[511111];
bool b[511111];
int n,m,s;
void spfa()
{

    int head=0,tail=1,i=0;
    for(int i=1; i<=n; i++) d[i]=2147483647;
    memset(b,0,sizeof(b));
    state[1]=s; b[s]=1;
    d[s]=0;
    do
    {

        i=list[state[++head]];
        while(i>0)
        {

            if(d[u[i]]+w[i]<d[v[i]])
            {

                d[v[i]]=d[u[i]]+w[i];
                if(!b[v[i]])
                {

                    b[v[i]]=1;
                    state[++tail]=v[i];

                }

            }
            i=next[i];

        }
        b[state[head]]=0;

    }while(tail!=head);

}
int main()
{

    scanf("%d%d%d",&n,&m,&s);
    for(int i=1; i<=m; i++)
    {

        scanf("%lld%lld%lld",&u[i],&v[i],&w[i]);
        next[i]=list[u[i]];
        list[u[i]]=i;

    }
    spfa();
    for(int i=1; i<=n; i++) printf("%lld ",d[i]);

}