某省自从实行了很多年的畅通工程计划后,终于修建了很多路。不过路多了也不好,每次要从一个城镇到另一个城镇时,都有许多种道路方案可以选择,而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。
现在,已知起点和终点,请你计算出要从起点到终点,最短需要行走多少距离。
Input
本题目包含多组数据,请处理到文件结束。
每组数据第一行包含两个正整数N和M(0<N<200,0<M<1000),分别代表现有城镇的数目和已修建的道路的数目。城镇分别以0~N-1编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0<=A,B<N,A!=B,0<X<10000),表示城镇A和城镇B之间有一条长度为X的双向道路。
再接下一行有两个整数S,T(0<=S,T<N),分别代表起点和终点。
Output
对于每组数据,请在一行里输出最短需要行走的距离。如果不存在从S到T的路线,就输出-1.
Sample Input
3 3 0 1 1 0 2 3 1 2 1 0 2 3 1 0 1 1 1 2
Sample Output
2 -1
好久没写spfa 复习一下
#include<bits/stdc++.h>
using namespace std;
const int maxn = 200 + 10;
int tot;
int head[maxn], vis[maxn], dis[maxn];
int n, m;
int S, t;
struct node{
int to;
int next;
int w;
node() {}
node(int a, int b, int c) : to(a), next(b), w(c) {}
}edge[maxn * maxn];
void init(){
memset(head, -1, sizeof(head));
memset(vis, 0, sizeof(vis));
memset(dis, 0x3f3f3f3f, sizeof(dis));
tot = 0;
}
edgeadd(int a, int b, int c){
edge[tot] = node(b, head[a], c);
head[a] = tot++;
edge[tot] = node(a, head[b], c);
head[b] = tot++;
}
void spfa(int s){
queue <int> q;
q.push(s);
vis[s] = 1;
dis[s] = 0;
while(!q.empty()){
int nw = q.front();
q.pop();
vis[nw] = 0;
for(int i = head[nw]; i != -1; i = edge[i].next){
int v = edge[i].to;
int w = edge[i].w;
if(dis[v] > w + dis[nw]){
dis[v] = w + dis[nw];
if(!vis[v]){
q.push(v);
vis[v] = 1;
}
}
}
}
if(dis[t] == 0x3f3f3f3f)
cout << "-1" << endl;
else
cout << dis[t] << endl;
}
int main(){
ios::sync_with_stdio(false);
while(cin >> n >> m){
init();
for(int i = 1; i <= m; i++){
int a, b, c;
cin >> a >> b >> c;
edgeadd(a, b, c);
}
cin >> S >> t;
spfa(S);
}
}