HDU 1874 Dijkstra算法

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

畅通工程续
Time Limit: 3000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 9293 Accepted Submission(s): 3099


Problem Description
某省自从实行了很多年的畅通工程计划后，终于修建了很多路。不过路多了也不好，每次要从一个城镇到另一个城镇时，都有许多种道路方案可以选择，而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。

现在，已知起点和终点，请你计算出要从起点到终点，最短需要行走多少距离。


Input
本题目包含多组数据，请处理到文件结束。
每组数据第一行包含两个正整数N和M(0<N<200,0<M<1000)，分别代表现有城镇的数目和已修建的道路的数目。城镇分别以0～N-1编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0<=A,B<N,A!=B,0<X<10000),表示城镇A和城镇B之间有一条长度为X的双向道路。
再接下一行有两个整数S,T(0<=S,T<N)，分别代表起点和终点。


Output
对于每组数据，请在一行里输出最短需要行走的距离。如果不存在从S到T的路线，就输出-1.


Sample Input
3 3
0 1 1
0 2 3
1 2 1
0 2
3 1
0 1 1
1 2


Sample Output
2
-1

#include <iostream>

using namespace std;

#define MAX_VERTEXT 250
const int MAX_WEIGHT= 0x7f7f7f7f;

int map[MAX_VERTEXT][MAX_VERTEXT];
int path[MAX_VERTEXT];

void Init()
{
    memset(map,MAX_WEIGHT,sizeof(map));
    memset(path,MAX_WEIGHT,sizeof(path));
}

void Input(int m)
{
    int start,end,weight;
    for (int i=0;i<m;i++)
    {
        cin>>start>>end>>weight;                    
        //这里太BT了。。。
        if (weight<map[start][end])
            map[start][end]=map[end][start]
                            =weight;
        if(start==end)
            map[start][end]=0;
    }
}

template<typename T>
T min(const T& x,const T& y)
{
    return x<y ? x:y;
}

int Dijkstra(int n,int s,int e)
{
    int i,j,k;
    int v[MAX_VERTEXT]={0};
    for (path[s]=0,i=0;i<n;i++)
    {
        //把顶点分成两个集合S，U（U为剩余顶点集）
        //一开始S中只有s顶点（开始点）
        //从剩余的顶点中挑选路径最短的加入S
        for(k=-1,j=0;j<n;j++)
            if(!v[j]&&(k==-1||path[j]<path[k]))
                k=j;

        for(v[k]=1,j=0;j<n;j++)
            if(!v[j]&&(map[k][j]!=MAX_WEIGHT))
                path[j]=::min(path[k]+map[k][j],path[j]);
    }
    if(path[e]>=MAX_WEIGHT)
        return -1;
    else
        return path[e];
}

int main()
{
    int N,M,start,end;
    while(cin>>N>>M)
    {
        Init();
        Input(M);
        cin>>start>>end;
        if(start==end)
            cout<<"0"<<endl;
        else
            cout<<Dijkstra(N,start,end)<<endl;
    }
    return 0;
}