题目描述
在每年的校赛里,所有进入决赛的同学都会获得一件很漂亮的t-shirt。但是每当我们的工作人员把上百件的衣服从商店运回到赛场的时候,却是非常累的!所以现在他们想要寻找最短的从商店到赛场的路线,你可以帮助他们吗?
Input
输入包括多组数据。每组数据第一行是两个整数N、M(N<=100,M<=10000),N表示成都的大街上有几个路口,标号为1的路口是商店所在地,标号为N的路口是赛场所在地,M则表示在成都有几条路。N=M=0表示输入结束。接下来M行,每行包括3个整数A,B,C(1<=A,B<=N,1<=C<=1000),表示在路口A与路口B之间有一条路,我们的工作人员需要C分钟的时间走过这条路。
输入保证至少存在1条商店到赛场的路线。
Output
对于每组输入,输出一行,表示工作人员从商店走到赛场的最短时间
Solution
直接套Dijkstra最短路算法求解即可
Code
#include<iostream>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
const int inf = 0x3f3f3f3f;
typedef struct Edge {
int to,cost;
};
vector<Edge> edges[105];
int dist[105];
bool visit[105];
int dijkstra(int n, int s, int t) {
memset(visit,0,sizeof(visit));
memset(dist,0x3f,sizeof(dist));
dist[s] = 0;
for (int i=1;i<=n;i++) {
int min_id=-1, min_dist=inf;
for (int j=1;j<=n;j++) {
if (!visit[j]&&dist[j]<min_dist) {
min_dist = dist[j];
min_id = j;
}
}
visit[min_id] = true;
for (auto& e: edges[min_id])
dist[e.to] = min(dist[e.to], dist[min_id]+e.cost);
}
return dist[t];
}
int main() {
int n,m;
int a,b,c;
while(cin>>n>>m) {
if (n==0&&m==0)
break;
for (int i=1;i<=n;i++)
edges[i].clear();
for (int i=1;i<=m;i++) {
cin>>a>>b>>c;
edges[a].push_back({b,c});
edges[b].push_back({a,c});
}
cout << dijkstra(n,1,n) << endl;
}
}