HDU 2544 最短路

http://acm.hdu.edu.cn/showproblem.php?pid=2544

 

Problem Description

在每年的校赛里，所有进入决赛的同学都会获得一件很漂亮的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

对于每组输入，输出一行，表示工作人员从商店走到赛场的最短时间

 

Sample Input

2 1

1 2 3

3 3

1 2 5

2 3 5

3 1 2

0 0

 

Sample Output

3

2

 

时间复杂度：$O(n^2)$

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<climits>
#include<queue>
#include<algorithm>
using namespace std;

#define N 110
#define MAX 999999
#define CLR(arr, what) memset(arr, what, sizeof(arr))

int nodenum, edgenum;
int map[N][N], dis[N];
bool visit[N];

int Dijkstra(int src, int des) {
    int temp, k;
    CLR(visit, false);
    for(int i = 1; i <= nodenum; ++i)
        dis[i] = (i == src ? 0 : map[src][i]);
    visit[src] = true;
    dis[src] = 0;
    for(int i = 1; i<= nodenum; ++i) {
        temp = MAX;
        for(int j = 1; j <= nodenum; ++j)
            if(!visit[j] && temp > dis[j])
                temp = dis[k = j];
        if(temp == MAX)
            break;
        visit[k] = true;
        for(int j = 1; j <= nodenum; ++j)
            if(!visit[j] && dis[j] > dis[k] + map[k][j])
                dis[j] = dis[k] + map[k][j];
    }
    return dis[des];
}

int main() {
    int start, end, cost;
    int answer;
    while(~scanf("%d%d", &nodenum, &edgenum) && (nodenum + edgenum)) {
        for(int i = 1; i <= nodenum; ++i)
            for(int j = 1; j <= nodenum; ++j)
            map[i][j] = MAX;
        for(int i = 1; i <= edgenum; ++i) {
            scanf("%d%d%d", &start, &end, &cost);
            if(cost < map[start][end])
                map[start][end] = map[end][start] = cost;
        }
        answer = Dijkstra(1, nodenum);
        printf("%d\n", answer);
    }
    return 0;
}