简单的实现了Dijkstra,不考虑效率问题。
注释:V表示图的顶点集合,S表示从源节点到集合中的顶点的最终最短路径已经确定的点的集合
Dijkstra算法的精髓就是
1)初始的时候源节点s到其他所有节点的距离都是无穷大。
2)将源节点加入到集合S,然后又该源节点发射出去所有边中,也就是所有能达到的节点,修改源节点s到这些节点的距离(这个就是算法的核心Relax操作)。
3)找出最短的距离的节点k,将该节点加入到S集合中,并且重复2)操作。直到所有节点被处理过了。
简要的说明一下Relax的作用。
程序中用DIS[u]来表示当前情况下源节点s到u的最短距离。那么对于一条边 u->v 而已,如果DIS[v] > DIS[u] + Weight(u,v), 那么就说明s点到v点最短距离可以更改为从s到u然后在到v。
原理可以参考《算法导论》在这里不详细说明了!
#include <iostream>
#include <limits>
using namespace std;
const int N = 5;
enum Color{ White, Gray, Black};
struct DEdge{
int start;
int end;
int weight;
};
int COST[N][N] = {{0 ,10 ,INT_MAX,5 ,INT_MAX},
{INT_MAX,0 ,1 ,INT_MAX,2 },
{INT_MAX,INT_MAX,0 ,INT_MAX,4 },
{INT_MAX,3 ,9 ,0 ,2 },
{7 ,INT_MAX,6 ,INT_MAX,0 }};
int PARENT[N];
int DIS[N];
Color COLOR[N];
void dijkstra_init(int source){
for(int i = 0 ; i < N ; ++i){
DIS[i] = INT_MAX;
COLOR[i] = White;
PARENT[i] = -1;
}
DIS[source] = 0;
COLOR[source] = Gray;
}
int findMinDist(){
int min = INT_MAX;
int pos = -1;
for(int i = 0 ; i < N; ++i)
if(COLOR[i]!=Black && min > DIS[i]){
min = DIS[i];
pos = i;
}
return pos;
}
void relax(int u, int v){
if(INT_MAX!= COST[u][v]){
if(DIS[v] > DIS[u]+COST[u][v]){
DIS[v] = DIS[u] + COST[u][v];
cout<<"set "<<v<<"="<<DIS[v]<<" through "<<u<<endl;
PARENT[v] = u;
}
}
for(int i = 0 ; i < N; ++i)
cout<<PARENT[i]<<' ';
cout<<endl;
}
void dijkstra(int source){
if(source>N) return;
dijkstra_init(source);
int count=0;
while(count<N){
int pos = findMinDist();
COLOR[pos] = Gray;
cout<<pos<<endl;
for(int i = 0; i < N; ++i)
relax(pos, i);
++count;
COLOR[pos] = Black;
}
for(int i = 0 ; i < N; ++i){
int pos = i;
cout<<pos;
while(PARENT[pos]!=-1){
cout<<"<-"<<PARENT[pos];
pos = PARENT[pos];
}
cout<<endl;
}
}
int main(){
dijkstra(0);
system("PAUSE");
return 0;
}