题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1874
注意:给Map数组赋值时,需要用循环赋值,用memset 赋值的话总是出错,WA了好几次,改了好久才AC
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
int Map[210][210];
int vis[210],dis[210];
int n,m;
void dij(int dk){
memset(vis,0,sizeof(vis));
for(int i=0;i<n;i++) dis[i]=Map[dk][i];
vis[dk]=1;
for(int i=1;i<n;i++){
int temp=INF,p=-1;
for(int j=0;j<n;j++){
if(!vis[j]&&temp > dis[j]){
temp=dis[j];
p=j;
}
}
if(temp==INF) break;
vis[p]=1;
for(int j=0;j<n;j++)
if(!vis[j])
dis[j]=min(dis[j],dis[p]+Map[p][j]);
}
}
int main()
{
int sx,ex,x,y,z;
while(~scanf("%d%d",&n,&m)){
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(i==j){
Map[i][j]=0;
}
else{
Map[i][j]=INF;
}
}
}
for(int i=0;i<m;i++){
scanf("%d%d%d",&x,&y,&z);
if(Map[x][y]>z){
Map[x][y]=z;
Map[y][x]=z;
}
}
scanf("%d%d",&sx,&ex);
dij(sx);
if(dis[ex]==INF) printf("-1\n");
else printf("%d\n",dis[ex]);
}
return 0;
}