hiho1093 ——spfa

hiho1093 ——spfa

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

万圣节的晚上，小Hi和小Ho在吃过晚饭之后，来到了一个巨大的鬼屋！

鬼屋中一共有N个地点，分别编号为1..N，这N个地点之间互相有一些道路连通，两个地点之间可能有多条道路连通，但是并不存在一条两端都是同一个地点的道路。

不过这个鬼屋虽然很大，但是其中的道路并不算多，所以小Hi还是希望能够知道从入口到出口的最短距离是多少？

提示：Super Programming Festival Algorithm。

输入

每个测试点（输入文件）有且仅有一组测试数据。

在一组测试数据中：

第1行为4个整数N、M、S、T，分别表示鬼屋中地点的个数和道路的条数，入口（也是一个地点）的编号，出口（同样也是一个地点）的编号。

接下来的M行，每行描述一条道路：其中的第i行为三个整数u_i, v_i, length_i，表明在编号为u_i的地点和编号为v_i的地点之间有一条长度为length_i的道路。

对于100%的数据，满足N<=10^5，M<=10^6, 1 <= length_i <= 10^3, 1 <= S, T <= N, 且S不等于T。

对于100%的数据，满足小Hi和小Ho总是有办法从入口通过地图上标注出来的道路到达出口。

输出

对于每组测试数据，输出一个整数Ans，表示那么小Hi和小Ho为了走出鬼屋至少要走的路程。

样例输入

5 10 3 5

1 2 997

2 3 505

3 4 118

4 5 54

3 5 480

3 4 796

5 2 794

2 5 146

5 4 604

2 5 63

样例输出

172
吐槽：图居然是无向的，题目没说清楚。。

#include<iostream>

#include<cstdio>

#include<cstring>

#include<cstdlib>

#include<algorithm>

#include<queue>



using namespace std;



typedef long long ll;

const int maxn=500010;

const ll INF=(1LL<<60);



int N,M,S,T;

bool vis[maxn];

ll dist[maxn];

int c[maxn];

struct Edge

{

    int v;

    ll w;

    Edge *next;

};Edge e[maxn*10];



bool relax(int u,int v,ll w)

{

    if(dist[u]+w<dist[v]){

        dist[v]=dist[u]+w;

        return true;

    }

    return false;

}



void add_edge(int u,int v,ll w) ///插入邻接表的首部而非尾部，避免遍历

{

    Edge *pre=&e[u];

    Edge *p=(Edge*)malloc(sizeof(Edge));

    p->v=v;p->w=w;

    p->next=pre->next;

    pre->next=p;

}



bool spfa()

{

    for(int i=1;i<=N;i++) dist[i]=(i==S)?0:INF;

    memset(vis,0,sizeof(vis));

    memset(c,0,sizeof(c));

    queue<int> q;

    q.push(S);

    vis[S]=1;c[S]++;

    while(!q.empty()){

        int u=q.front();q.pop();vis[u]=0;

        for(Edge *p=e[u].next;p!=NULL;p=p->next){ ///遍历队首结点的每条出边

            int v=p->v;

            ll w=p->w;

            if(relax(u,v,w)){

                if(!vis[v]){

                    q.push(v);

                    vis[v]=1;

                    c[v]++;

                    if(c[v]>N) return false;

                }

            }

        }

    }

    return true;

}



int main()

{

    cin>>N>>M>>S>>T;

    memset(e,0,sizeof(e));

    while(M--){

        int u,v,w;

        scanf("%d%d%d",&u,&v,&w);

        add_edge(u,v,w);

        add_edge(v,u,w);

    }

    spfa();

    cout<<dist[T]<<endl;

    return 0;

}

View Code