Dijkstra算法是解决单源最短路的经典算法,现对这样一道简单的题目给出完整代码。
输入n和m,代表n个节点,m条边,然后是m行输入,每行有x,y,z,代表x到y的路距离为z。问题:从1出发到各点的最短路径。
测试样例:
/*
10 12
1 4 1
1 5 1
1 6 1
4 8 1
4 3 1
3 5 1
5 7 1
3 7 1
6 2 1
7 2 1
7 10 1
2 9 1
*/
直接给出AC代码。
#include<cstdio>
#include<cstring>
const int maxn=100;
int map[maxn][maxn];
int dis[maxn];
int path[maxn];
bool vis[maxn];
int n;
void dijk(int s)
{
memset(path,-1,sizeof(path));
memset(dis,0x3f,sizeof(dis)); //快速初始化为无穷大
memset(vis,0,sizeof(vis));
dis[s]=0;
while(1)
{
int k=0;
for(int j=1;j<=n;j++)
{
if(!vis[j]&&dis[j]<dis[k]) //这一步找未收录顶点中dis值最小的
k=j;
}
if(!k) return ; //没有未收录的顶点,则返回(注意顶点从1开始,k为0一定是没找到)
vis[k]=1; //收录顶点k
for(int j=1;j<=n;j++)
{
if(dis[j]>dis[k]+map[k][j])
{
dis[j]=dis[k]+map[k][j];
path[j]=k;
}
}
}
}
void print(int x)
{
if(x==-1) return ;
print(path[x]);
printf("%d->",x);
}
int main()
{
int m,x,y,z,order;
scanf("%d%d",&n,&m);
memset(map,0x3f,sizeof(map));
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&x,&y,&z);
map[x][y]=map[y][x]=z;
}
dijk(1);
scanf("%d",&order);
printf("%d\n",dis[order]);
print(path[order]);
printf("%d",order);
return 0;
}