王道机试练习——求最短距离Dijkstra算法
题目描述
在每年的校赛里,所有进入决赛的同学都会获得一件很漂亮的 t-shirt。但是 每当我们的工作人员把上百件的衣服从商店运回到赛场的时候,却是非常累 的 ! 所以现在他们想要寻找最短的从商店到赛场的路线,你可以帮助他们吗?
输入:
输入包括多组数据。每组数据第一行是两个整数 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 条商店到赛场的路线。 当输入为两个 0 时,输入结束。
输出: 对于每组输入,输出一行,表示工作人员从商店走到赛场的最短时间。
代码
#include<stdio.h>
#include<vector>
using namespace std;
struct E {
int next;//直接相邻接点
int c;//该边权值
};
vector<E>edge[101];//邻接链表
bool mark[101];
bool Dis[101];
int main() {
int n, m;
while (scanf("%d%d", &n, &m) != EOF) {
if (n == 0 || m == 0) break;
for (int i = 1; i <= n; i++) edge[i].clear();
while (m--) {
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
E tmp;
tmp.c = c;
tmp.next = b;
edge[a].push_back(tmp);
tmp.next = a;
edge[b].push_back(tmp);
}
for (int i = 1; i <= n; i++) {
Dis[i] = -1;
mark[i] = false;
}
Dis[1] = 0;
mark[1] = true;
int newP = 1;
for (int i = 1; i < n; i++) {
for (int j = 0; j < edge[newP].size(); j++) {
int t = edge[newP][j].next;
int c = edge[newP][j].c;
if (mark[t] == true) continue;
if (Dis[t] == -1 || Dis[t] > Dis[newP] + c) {
Dis[t] = Dis[newP] + c;
}
}
int min = 123123123;
for (int j = 1; j <= n; j++) {
if (mark[j] == true) continue;
if (Dis[j] == -1) continue;
if (Dis[j] < min) {
min = Dis[j];
newP = j;
}
}
mark[newP] = true;
}
printf("%d\n", Dis[n]);
}
return 0;
}