我的不知道对不对的dij模板

来源:互联网 发布:nginx 跳转到二级目录 编辑:程序博客网 时间:2024/05/21 09:12

慢慢改吧 谁看到bug还希望好心指出来

#include <iostream>
#include <cstdio>
#include <cstring>
#include <climits>
#include <queue>
#define MAXN 100
#define MAXM 10000
#define inf (1<<30)
using namespace std;
int d[MAXN],r[MAXN][MAXN],fa[MAXN],path[MAXM],n,m;
bool visi[MAXM];//点是否访问过
void read()
{
    int s,e,v;
    scanf("%d %d",&n,&m);//输入点数与路数
    memset(r,0,sizeof(r));
    for(int i=1;i<=m;++i)
    {
        scanf("%d %d %d",&s,&e,&v);
        r[s][e]=v;
        //无向的话为r[s][e]=r[e][s]=v;
    }
    for(int i=1;i<=n;++i)
        r[i][i]=inf;
    for(int i=1;i<=n;++i)
        d[i]=inf;
    memset(visi,0,sizeof(visi));
}
int dij(int s,int e)
{
    d[s]=0,fa[s]=-1;
    for(int i=1;i<=n;++i)
    {
        int x=0,u=inf;
        for(int j=1;j<=n;++j)
            if(!visi[j]&&d[j]<=u)
                u=d[x=j];
        visi[x]=true;
        if(x==0) break;
        for(int j=1;j<=n;++j)
        {
            if(!visi[j]&&d[j]>d[x]+r[x][j])
            {
                d[j]=d[x]+r[x][j];
                fa[j]=x;
            }
        }
    }
    return d[e];
}
void spath(int e)
{
    int t=e,cnt=0;
    while(t!=-1)
    {
        path[cnt++]=t;
        t=fa[t];
    }
    for(int i=cnt-1;i>=0;--i)
        printf("%d ",path[i]);
    printf("\n");
}


int main()
{
    read();
    dij(1,n);
    spath(n);


    return 0;
}


0 0
原创粉丝点击