hihocoder#1093 : 最短路径·三：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<bits/stdc++.h>
using namespace std;
const int MAX=2e6;
const int INF=1e9+7;
struct lenka
{
    int s,ds;
    int operator<(const lenka& x)const{return x.ds<ds;}
};
struct EDG
{
    int to,next,val;
}ed[MAX];
int head[MAX],v[MAX],tot=0;
int n,d[MAX];
void add(int x,int y,int z)
{
    ed[tot].to=y;
    ed[tot].val=z;
    ed[tot].next=head[x];
    head[x]=tot++;
}
void spfa(int s)
{
    memset(v,0,sizeof v);
    for(int i=1;i<=n;i++)d[i]=INF;
    d[s]=0;
    priority_queue<lenka>p;
    p.push((lenka){s,0});
    while(!p.empty())
    {
        lenka now=p.top();p.pop();
        if(v[now.s])continue;
        v[now.s]=1;
        for(int i=head[now.s];i!=-1;i=ed[i].next)
        {
            int nex=ed[i].to;
            if(d[nex]>now.ds+ed[i].val)
            {
                d[nex]=now.ds+ed[i].val;
                p.push((lenka){nex,d[nex]});
            }
        }
    }
}
int main()
{
    int m,s,t;
    memset(head,-1,sizeof head);
    scanf("%d%d%d%d",&n,&m,&s,&t);
    while(m--)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
    spfa(s);
    printf("%d\n",d[t]);
    return 0;
}