首页 > 代码库 > Dijkstra模板题图论书p133

Dijkstra模板题图论书p133

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <climits>
#include <string>
#include <iostream>
#include <map>
#include <cstdlib>
#include <list>
#include <set>
#include <queue>
#include <stack>
using namespace std;
#define maxn 1000
#define INF 100000000
int edge[maxn][maxn];
int n,s[maxn],path[maxn],dist[maxn];
void Dijkstra(int v0)
{
    int i,j,k;
    for(i=0;i<n;i++)
    {
        dist[i]=edge[v0][i];s[i]=0;
        if(i!=v0&&dist[i]<INF) path[i]=v0;
        else path[i]=-1;
    }
    s[v0]=1;dist[v0]=0;
    for(i=0;i<n-1;i++)
    {
        int min=INF,u=v0;
        for(j=0;j<n;j++)
        {
            if(!s[j]&&dist[j]<min)
            {
                min=dist[j];
                u=j;
            }
        }
        s[u]=1;
        for(k=0;k<n;k++)
        {
            if(!s[k]&&edge[u][k]+dist[u]<dist[k])
            {
                dist[k]=edge[u][k]+dist[u];
                path[k]=u;
            }
        }
    }
}
int main()
{
    int i,j;
    int u,v,w;
    scanf("%d",&n);
    while(1)
    {
        scanf("%d%d%d",&u,&v,&w);
        if(u==-1&&v==-1&&w==-1) break;
        edge[u][v]=w;
    }
    for(i=0;i<n;i++)
    {
        for(j=0;j<n;j++)
        {
            if(i==j) edge[i][j]=0;
            else if(edge[i][j]==0) edge[i][j]=INF;
        }
    }
    Dijkstra(0);
    int shortest[maxn];
    for(i=1;i<n;i++)
    {
        printf("%d\t",dist[i]);
        memset(shortest,0,sizeof(shortest));
        int k=0;
        shortest[k]=i;
        while(path[shortest[k]]!=0)
        {
            k++;
            shortest[k]=path[shortest[k-1]];
        }
        k++;
        shortest[k]=0;
        for(j=k;j>0;j--)
        {
            printf("%d->",shortest[j]);
        }
        printf("%d\n",shortest[0]);
    }
    return 0;
}