Bellman-ford Spfa hihocoder1903

时间限制: 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<cmath>
#include<vector>
#include<queue>
using namespace std;
#define MAXN 1000100
#define inf 1LL<<60

struct Edge
{
    int v,w,next;
} edge1[MAXN*10],edge2[MAXN*10];

int head1[MAXN],head2[MAXN];
long long dist[MAXN];
bool mark[MAXN];
int n,m,NE;

void Insert(Edge *edge,int *head,int u,int v,int w)
{
    edge[NE].v=v;
    edge[NE].w=w;
    edge[NE].next=head[u];
    head[u]=NE++;

    edge[NE].v=u;
    edge[NE].w=w;
    edge[NE].next=head[v];
    head[v]=NE++;
}


long long SPFA(Edge *edge,int *head,int u)
{
    memset(mark,false,(n+2)*sizeof(bool));
    for(int i=1; i<=n; i++)dist[i]=inf;
    dist[u]=0;
    queue<int>Q;
    Q.push(u);
    while(!Q.empty())
    {
        u=Q.front();
        Q.pop();
        mark[u]=false;
        for(int i=head[u]; i!=-1; i=edge[i].next)
        {
            int v=edge[i].v,w=edge[i].w;
            if(dist[u]+w<dist[v])
            {
                dist[v]=dist[u]+w;
                if(!mark[v])
                {
                    mark[v]=true;
                    Q.push(v);
                }
            }
        }
    }
    long long ans=0;
    for(int i=2; i<=n; i++)ans+=dist[i];
    return ans;
}

int main()
{
    int u,v,w;
    int S, T;
    while(~scanf("%d%d%d%d",&n,&m, &S, &T))
    {
        NE=0;
        memset(head1,-1,(n+2)*sizeof(int));
        memset(head2,-1,(n+2)*sizeof(int));
        while(m--)
        {
            scanf("%d%d%d",&u,&v,&w);
            Insert(edge1,head1,u,v,w);
            //Insert(edge2,head2,v,u,w);//建反图
            //NE++;
        }
        SPFA(edge1,head1,S);
        printf("%lld\n",dist[T]);
    }
    //printf("%lld\n",SPFA(edge1,head1,1)+SPFA(edge2,head2,1));

    return 0;
}