图论——单源最短路径算法之Dijkstra算法

来源:互联网 发布:软件系统试运行方案 编辑:程序博客网 时间:2024/06/05 20:07

简单的实现了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;}


原创粉丝点击