hdu 2544 最短路 題解 （dijkstra/迪傑斯特拉算法）

題目鏈接：http://acm.hdu.edu.cn/showproblem.php?pid=2544

 

這道題用dijkstra模板一套就出來了。

模板：http://blog.csdn.net/xdz78/article/details/47719849

 

需要注意的是，這裏的邊應該是雙向邊，所以在輸入邊的數據的時候應該這樣寫：

for(i=0;i<m;i++){
            scanf("%d%d%d",&a,&b,&c);
            g.map[a][b]=g.map[b][a]=c;
        }

我就是這裏錯了所以樣例都過不了，後來一問才發現的。

 

代碼：

#include <iostream>
#include <cstdio>
#include <cstring>
#define N 110

using namespace std;

typedef struct node {
    int map[N][N];//鄰接矩陣
    int n;//頂點數
    int m;//邊數
}MGragh;

void dijkstra(int *pre,int *dis,MGragh g,int v0){
    int vis[g.n+1];
    int i,j,k;
    memset(vis,0,sizeof(vis));
    for(i=1;i<=g.n;i++){
        if(g.map[v0][i]>0&&i!=v0){
            dis[i]=g.map[v0][i];
            pre[i]=v0;
        }
        else {
            dis[i]=INT_MAX;
            pre[i]=-1;
        }
        vis[i]=false;
        dis[v0]=0;
        pre[v0]=v0;
    }
    vis[v0]=true;
    for(i=1;i<=g.n;i++){
        int min=INT_MAX;
        int u;
        for(j=1;j<=g.n;j++){
            if(min>dis[j]&&vis[j]==false){
                min=dis[j];
                u=j;
            }
        }
        vis[u]=true;
        for(k=1;k<=g.n;k++){
            if(vis[k]==false&&g.map[u][k]>0&&g.map[u][k]+min<dis[k]){
                dis[k]=g.map[u][k]+min;
                pre[k]=u;
            }
        }
    }
}

int main()
{
    int i,j,a,b,c;
    int n,m;
    while(scanf("%d%d",&n,&m)&&n+m>0){
        MGragh g;
        g.n=n;//初始化
        g.m=m;
        int pre[n+1];
        int dis[n+1];
        for(i=0;i<N;i++){
            for(j=0;j<N;j++){
                g.map[i][j]=0;
            }
        }
        memset(pre,0,sizeof(pre));
        memset(dis,0,sizeof(dis));
        for(i=0;i<m;i++){
            scanf("%d%d%d",&a,&b,&c);
            g.map[a][b]=g.map[b][a]=c;
        }
        dijkstra(pre,dis,g,1);
        printf("%d\n",dis[n]);
    }
    return 0;
}