1. 程式人生 > >迪傑斯特拉/dijkstra 算法模板(具體凝視)

迪傑斯特拉/dijkstra 算法模板(具體凝視)

定義 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 算法模板(具體凝視)