迪傑斯特拉/dijkstra 算法模板(具體凝視)
阿新 • • 發佈:2017-05-31
定義 ostream all popu post 路徑 ret typedef tdi
#include <iostream> #include <malloc.h> #include <cstring> #include <stack> #include <cstdio> //定義鄰接矩陣的大小 #define N 100 #define M 100 using namespace std; typedef struct node { int map[N][M];//鄰接矩陣 int n;//頂點數 int e;//邊數 }MGragh; void dijkstra(MGragh g,int *dis,int *pre,int v0){ int i,j,k; bool *visited=(bool *)malloc(sizeof(bool)*g.n);//標記數組 for(i=0;i<g.n;i++){//初始化 if(g.map[v0][i]>0&&i!=0){ dis[i]=g.map[v0][i]; pre[i]=v0; } else{ dis[i]=INT_MAX; pre[i]=-1; } pre[v0]=v0; dis[v0]=0; } visited[v0]=true;//標記源點v0為訪問過 for(i=1;i<g.n;i++){//執行n-1次 int min=INT_MAX;//初始化 int u; for(j=0;j<g.n;j++){//尋找未訪問過的頂點中權值最小的那個 if(visited[j]==false&&dis[j]<min){ min=dis[j];//記錄下來 u=j; } } visited[u]=true;//標記為訪問過 //更新dis數組的值和路徑 for(k=0;k<g.n;k++){ if(visited[k]==false&&g.map[u][k]>0&&g.map[u][k]+min<dis[k]){ dis[k]=min+g.map[u][k]; pre[k]=u; } } } } void showpath(int *pre,int v,int v0){//v是當前節點。 //輸出源點v0到當前節點v的路徑 stack<int>s; while(v!=v0){ s.push(v); v=pre[v]; } s.push(v); while(!s.empty()){ printf("%d ",s.top()); s.pop(); } } int main() { int n,e; while(scanf("%d%d",&n,&e)&&e!=0){ int i,j; int s,t,w;//起始點s,終點t,邊st的權值為w MGragh g; int v0;//源點v0 int *dis=(int *)malloc(sizeof(int)*n);//dis[i]記錄從源點v0到當前點i的路徑長度 int *pre=(int *)malloc(sizeof(int)*n);//記錄每一個點的前驅,即pre[i]=j;說明i點的前驅為j for(i=0;i<N;i++){//初始化 for(j=0;j<M;j++){ g.map[i][j]=0; } } g.n=n; g.e=e; for(i=0;i<e;i++){//建立鄰接矩陣 scanf("%d%d%d",&s,&t,&w); g.map[s][t]=w; } scanf("%d",&v0); dijkstra(g,dis,pre,v0); for(i=0;i<n;i++){ if(i!=v0){ showpath(pre,i,v0);//輸出路徑 printf("%d\n",dis[i]);//輸出最短路徑的大小 } } } return 0; } /* 測試用例 5 7 0 1 100 0 2 30 0 4 10 2 1 60 2 3 60 3 1 10 4 3 50 0 */
迪傑斯特拉/dijkstra 算法模板(具體凝視)