Light oj 1002 Country Roads (Dijkstra)

题目连接：

　　http://www.lightoj.com/volume_showproblem.php?problem=1002

题目描述：

　　有n个城市，从0到n-1开始编号，n个城市之间有m条边，中心城市为t，问每个城市到中心城市的最小路径的花费，路径花费大小的定义为：一条路上花费最大的边的值。

解题思路：

　　Dijkstra的变形，用Dijkstra求出来的单源路径可以保证每条边都是最优的，所以最短路上的最长边就是所求。

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

const int maxn = 510;
const int INF = 0x3f3f3f3f;
int map[maxn][maxn], dist[maxn];
void init ()
{
    int i, j;
    for (i=0; i<maxn; i++)
        for (j=0; j<maxn; j++)
            if (i==j)
                map[i][j] = 0;
            else
                map[i][j] = INF;
}
void dijkstra (int s, int n)
{
    int i, j, vis[maxn];
    memset (vis, 0, sizeof(vis));

    for (i=0; i<n; i++)
        dist[i] = map[s][i];
    vis[s] = 1;

    for (i=1; i<n; i++)
    {
        int mini = INF, index;
        for (j=0; j<n; j++)
            if (!vis[j] && mini > dist[j])
            {
                mini = dist[j];
                index = j;
            }
            if (mini == INF)
                return ;
        vis[index] = 1;
        for (j=0; j<n; j++)
            if (!vis[j])
            {
                dist[j] = min(dist[j], max(map[index][j],dist[index]));//重点
            }
    }
}
int main ()
{
    int T, n, m, l = 1;
    scanf ("%d", &T);
    while (T --)
    {
        init ();
        scanf ("%d %d", &n, &m);
        while (m --)
        {
            int u, v, w;
            scanf ("%d %d %d", &u, &v, &w);
            if (map[u][v] > w)//有重边，选最优
            map[u][v] = map[v][u] = w;
        }
        int t;
        scanf ("%d", &t);
        dijkstra (t, n);
        printf ("Case %d:
", l ++);
        for (int i=0; i<n; i++)
            if (dist[i] != INF)
                printf ("%d
", dist[i]);
            else
                printf ("Impossible
");
    }
    return 0;
}