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;
}