[hihoCoder] #1081 : 最短路径·一

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

万圣节的早上，小Hi和小Ho在经历了一个小时的争论后，终于决定了如何度过这样有意义的一天——他们决定去闯鬼屋！

在鬼屋门口排上了若干小时的队伍之后，刚刚进入鬼屋的小Hi和小Ho都颇饥饿，于是他们决定利用进门前领到的地图，找到一条通往终点的最短路径。

鬼屋中一共有N个地点，分别编号为1..N，这N个地点之间互相有一些道路连通，两个地点之间可能有多条道路连通，但是并不存在一条两端都是同一个地点的道路。那么小Hi和小Ho至少要走多少路程才能够走出鬼屋去吃东西呢？

提示：顺序！顺序才是关键。

输入

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

在一组测试数据中：

第1行为4个整数N、M、S、T，分别表示鬼屋中地点的个数和道路的条数，入口（也是一个地点）的编号，出口（同样也是一个地点）的编号。

接下来的M行，每行描述一条道路：其中的第i行为三个整数u_i, v_i, length_i，表明在编号为u_i的地点和编号为v_i的地点之间有一条长度为length_i的道路。

对于100%的数据，满足N<=10^3，M<=10^4, 1 <= length_i <= 10^3, 1 <= S, T <= N, 且S不等于T。

对于100%的数据，满足小Hi和小Ho总是有办法从入口通过地图上标注出来的道路到达出口。

输出

对于每组测试数据，输出一个整数Ans，表示那么小Hi和小Ho为了走出鬼屋至少要走的路程。

样例输入

5 23 5 4
1 2 708
2 3 112
3 4 721
4 5 339
5 4 960
1 5 849
2 5 98
1 4 99
2 4 25
2 1 200
3 1 146
3 2 106
1 4 860
4 1 795
5 4 479
5 4 280
3 4 341
1 4 622
4 2 362
2 3 415
4 1 904
2 1 716
2 5 575

样例输出

123

就是Dijkstra算法，之前写得比较少，各种经典的图算法还是得自己码一遍才行啊。

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <climits>
using namespace std;

int N, M, S, T;
vector<vector<int> > m(1001, vector<int>(1001));
vector<int> dist(1001);
vector<int> s(1001);

void solve() {
    dist.assign(N+1, INT_MAX);
    s.assign(N+1, false);
    for (int i = 1; i <= N; ++i) {
        dist[i] = m[S][i];
    }
    dist[S] = 0; s[S] = true;
    for (int i = 1; i <= N; ++i) {
        int tmp_min = INT_MAX;
        int u = S;
        //从未用过的点中找到距离最小的点
        for (int j = 1; j <= N; ++j) {
            if (!s[j] && dist[j] < tmp_min) {
                u = j;
                tmp_min = dist[j];
            }
        }
        s[u] = true;
        //更新距离
        for (int j = 1; j <= N; ++j) {
            if (!s[j] && m[u][j] < INT_MAX) {
                dist[j] = min(dist[j], dist[u]+m[u][j]);
            }
        }
    }
    cout << dist[T] << endl;
}

int main() {
    while (cin >> N >> M >> S >> T) {
        int u, v, len;
        for (int i = 1; i <= N; ++i)
            m[i].assign(N+1, INT_MAX);
        for (int i = 0; i < M; ++i) {
            cin >> u >> v >> len;
            m[u][v] = m[v][u] = min(m[u][v], len);
        }
        solve();
    }
    return 0;
}