迪杰斯特拉/dijkstra 算法模板(具体凝视)
#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 */