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
代码:迪杰斯特拉dis[]记录的是起点到每一点的最短距离,
不断更新,而普利姆
dis[]记录的是原集合体(树)和剩余每一点的距离。
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#define N 400
#define inf 999999
int map[N][N];
int dis[N];
int vis[N];
int s,e;
void dijkstra(int n)
{
int i,j;
int pos;
for(i=0; i<n; i++)
dis[i]=map[s][i];
vis[s]=1;
for(i=1; i<n; i++)
{
int min=inf;
for(j=0; j<n; j++)
if(!vis[j]&&dis[j]<min)
{
min=dis[j];
pos=j;
}
vis[pos]=1;
for(j=0; j<n; j++)
if(!vis[j]&&dis[j]>dis[pos]+map[j][pos]) //迪杰斯特拉很像最小生
dis[j]=dis[pos]+map[j][pos]; //成树的普利姆,就是这里不一样。
}
}
int main()
{
int n,m;
int i,j;
int a,b,x;
while(scanf("%d %d",&n,&m)!=EOF)
{
memset(vis,0,sizeof(vis));
for(i=0; i<n; i++)
{for(j=0; j<n; j++)
{map[i][j]=inf;}
map[i][i]=0; //这样写更严密,也更符合实际,自己到自己距
} //离为0,不写有可能不过。
for(i=1; i<=m; i++)
{
scanf("%d %d %d",&a,&b,&x);
if(map[a][b]>x)
map[a][b]=map[b][a]=x;
}
scanf("%d %d",&s,&e);
dijkstra(n);
if(dis[e]!=inf) //注意输出的判断。
printf("%d\n",dis[e]);
else
printf("-1\n");
}
return 0;
}