hdu1874(暢通工程續)

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

#include<iostream>
#define _max 9999999
#define M 201
using namespace std;
int map[M][M];
int visit[M];
int dir[M];
int n , m ;
int a , b , c ;
int len;
void Init(int n)
{
int i,j;
for(i = 0 ; i <n ; i++)
{
for(j=0 ; j <n; j++)
map[i][j] = _max;
}
}
void DJ(int v)
{
int i,j,p,k ,_min;
for(i=0;i<=n;i++)
dir[i]=map[v][i];
dir[v]=0;
visit[v]=true;
for(i=0;i<=n;i++)
{
_min=_max;
for(j=0;j<n;j++)
{
if(!visit[j] && _min>dir[j])
{
_min = dir[j];
k = j;
}
}
visit[k] = true ;
for( p=0 ; p<n; p++)
{
if(!visit[p] && dir[p]>dir[k] + map[k][p])
{
dir[p] = dir[k] + map[k][p];
}
}
}
}
int main()
{

while(cin>>n>>m)
{
Init(n);
int i;
int s,e;
for(i=0;i<m;i++)
{
cin>>a>>b>>c;
if(map[a][b]>c) map[a][b] = map[b][a] = c;

}
memset(visit , false , sizeof(visit));
cin>>s>>e;
DJ(s);
if(dir[e]<_max)cout<<dir[e]<<endl;
else cout<<"-1"<<endl;
}
return 0;

}