最短路径 （Dijstra）与prim的相似

今天好好的，仔细的研究了下Dijstra定理，现在有了比较深的理解。因此模仿写下了这个函数

const int MAXSIZE=100; const int INF=1000000;//除对角线外最初都要初始化为无穷 int dist[MAXSIZE]; void Dijkstra(int cost[][MAXSIZE],int n,int v)//求v到各个顶点的最短路径 { int i,j,k,min,set[MAXSIZE]; memset(set,0,sizeof(set));初始化 set[v]=1; for(i=0;i<n;i++) dist[i]=cost[v][i]; for(i=0;i<n;i++) { min=INF; k=-1; for(j=0;j<n;j++) if(dist[j]<min&&set[j]!=1)/*选取不在set中且具有最小距离的顶点k*/ { min=dist[j]; k=j; } set[k]=1; for(j=0;j<n;j++) { /*修改不在s中的顶点的距离*/ if(cost[k][j]!=INF&&dist[j]>(dist[k]+cost[k][j])&&set[j]!=1) dist[j]=dist[k]+cost[k][j]; }//dist数组存的就是v到个点的最短路径 } }

虽然接受速度有点慢……但还是挺高兴的，终于理解了…… 

而且发现Dijstra与prim在写法上有很大相似之处，所以我把prim算法的代码也贴出来

const int MAXSIZE=100; const int INF=1000000;//对角线外最初都要初始化为无穷 int lowcost[MAXSIZE]; void prim(int cost[][MAXSIZE],int n,int v)//从任意点v出发生成最小树 { int i,j,k,min,set[MAXSIZE]; memset(lowcost,INF,sizeof(lowcost));//初始化为无穷 memset(set,0,sizeof(set)); set[v]=1; for(i=0;i<n;i++) dist[i]=cost[v][i]; for(i=0;i<n;i++) { min=INF; k=-1; for(j=0;j<n;j++) if(lowcost[j]<min&&set[j]!=1)/*选取不在lowcost中具有最小距离的顶点k*/ { min=dist[j]; k=j; } set[k]=1; for(j=0;j<n;j++) { if(lowcost[j]>cost[k][j]&&set[j]!=1) lowcost[j]=cost[k][j]; }//lowcost数组存的就是v到各个点的最短路径权值 } }

下面是我老师写的：

 

#include <iostream> #include <fstream> /*input.txt 6 9 6 1 5 9 9 6 9 5 9 3 9 1 5 9 5 6 4 5 9 5 9 9 2 9 3 6 9 9 6 9 9 4 2 6 9 */ using namespace std; #define NUM 100 int graph[NUM][NUM]; int max=10; void prim(int graph[][NUM],int num,int begin) { int i,j,min,mink; int lowcost[NUM]; int adjvex[NUM]; for(i=1;i<num;i++) { lowcost[i]=graph[begin][i]; adjvex[i]=begin; } //找最小边 for(j=1;j<num;j++) { min=lowcost[1]; mink=1; for(i=2;i<num;i++) if(lowcost[i]<min) { min=lowcost[i]; mink=i; } cout<<mink+1<<"->"<<adjvex[mink]+1<<" "<<min<<endl; lowcost[mink]=max; //调整lowcost for(i=1;i<num;i++) { if(lowcost[i]>graph[i][mink]&&lowcost[i]<max) { lowcost[i]=graph[i][mink]; adjvex[i]=mink; } } } } int main() { ifstream cin1("input.txt"); int i,j; int num; cin1>>num; for(i=0;i<num;i++) for(j=0;j<num;j++) cin1>>graph[i][j]; for(i=0;i<num;i++) { for(j=0;j<num;j++) cout<<graph[i][j]<<" "; cout<<endl; } prim(graph,num,0); return 0; }