hihoCoder - 1093 - 最短路径·三：SPFA算法

1093 : 最短路径·三：SPFA算法

时间限制:10000ms 单点时限:1000ms 内存限制:256MB

描述

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

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

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

输入

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

在一组测试数据中：

第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<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define INF 0xffffff
#define MAXN 100005
using namespace std;

int dis[MAXN];
int vis[MAXN];
int head[MAXN];
int n, m, s, t, cnt;

struct edge{
int v, w, next;
}edge[1000005];

void addedge(int u, int v, int w){
edge[cnt].v = v;
edge[cnt].w = w;
edge[cnt].next = head[u];
head[u] = cnt++;
}

void spfa(int s){
int pos, v;
queue<int> q;
q.push(s);
dis[s] = 0;
vis[s] = 1;
while(!q.empty()){
pos = q.front();
q.pop();
vis[pos] = 0;

for(int i = head[pos]; i!=-1; i = edge[i].next){
v = edge[i].v;
if(dis[pos]+edge[i].w < dis[v]){
dis[v] = dis[pos]+edge[i].w;
if(!vis[v]){
vis[v] = 1;
q.push(v);
}
}
}
}
}

int main()
{
while(scanf("%d %d %d %d", &n, &m, &s, &t)!=EOF){
cnt = 0;
memset(head, -1, sizeof(head));
memset(vis, 0, sizeof(vis));
memset(dis, 0x3f, sizeof(dis));

for(int i = 0; i < m; i++){
int u, v, w;
scanf("%d %d %d", &u, &v, &w);
addedge(u, v, w);
addedge(v, u, w);
}

spfa(s);

printf("%d\n", dis[t]);
}
return 0;
}