哈哈
OK,现在来总结一下刚才的算法。算法的基本思想是:每次找到离源点(上面例子的源点就是 1 号顶点)最近的一个顶点,然后以该顶点为中心进行扩展,最终得到源点到其余所有点的最短路径。基本步骤如下:
- 将所有的顶点分为两部分:已知最短路程的顶点集合 P 和未知最短路径的顶点集合 Q。最开始,已知最短路径的顶点集合 P 中只有源点一个顶点。我们这里用一个 book[ i ]数组来记录哪些点在集合 P 中。例如对于某个顶点 i,如果 book[ i ]为 1 则表示这个顶点在集合 P 中,如果 book[ i ]为 0 则表示这个顶点在集合 Q 中。
- 设置源点 s 到自己的最短路径为 0 即 dis=0。若存在源点有能直接到达的顶点 i,则把 dis[ i ]设为 e[s][ i ]。同时把所有其它(源点不能直接到达的)顶点的最短路径为设为 ∞。
- 在集合 Q 的所有顶点中选择一个离源点 s 最近的顶点 u(即 dis[u]最小)加入到集合 P。并考察所有以点 u 为起点的边,对每一条边进行松弛操作。例如存在一条从 u 到 v 的边,那么可以通过将边 u->v 添加到尾部来拓展一条从 s 到 v 的路径,这条路径的长度是 dis[u]+e[u][v]。如果这个值比目前已知的 dis[v]的值要小,我们可以用新值来替代当前 dis[v]中的值。
- 重复第 3 步,如果集合 Q 为空,算法结束。最终 dis 数组中的值就是源点到所有顶点的最短路径。
完整的 Dijkstra 算法代码如下:
可以输入以下数据进行验证。第一行两个整数 n m。n 表示顶点个数(顶点编号为 1~n),m 表示边的条数。接下来 m 行表示,每行有 3 个数 x y z。表示顶点 x 到顶点 y 边的权值为 z。
6 9
1 2 1
1 3 12
2 3 9
2 4 3
3 5 5
4 3 4
4 5 13
4 6 15
5 6 4
运行结果是
0 1 8 4 13 17
有个奇异的事情就是奇诡的scanf和cin的问题,是时候看看cin cout流输入的特点了
#include <iostream>
#include<cstdio>
using namespace std;
int INF=999999;
const int maxn=300;
int book[maxn];
int dis[maxn];
int e[maxn][maxn];
int m,n;
int main()
{
cin>>n>>m;
int x,y,d;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{ if(i==j)
e[i][j]=0;
else
e[i][j]=INF;
}
for(int i=1;i<=m;i++)
{
scanf("%d %d%d",&x,&y,&d);
e[x][y]=d;
}
for(int i=1;i<=n;i++)
{
book[i]=0;
}
for(int i=1;i<=n;i++)
dis[i]=e[1][i];
book[1]=1;
int minn,next=0;
for(int i=1;i<=n-1;i++)
{
minn=INF;
for(int j=1;j<=n;j++)
{
if(book[j]==0&&dis[j]<minn)
{
minn=dis[j];
next=j;
}
}
book[next]=1;
for(int j=1;j<=n;j++)
{
if(e[next][j]<INF)
{
if(dis[j]>dis[next]+e[next][j])
dis[j]=dis[next]+e[next][j];
}
}
}
for(int i=1;i<=n;i++)
printf("%d ",dis[i]);
return 0;
}