poj 2378 (dijkstra)

http://poj.org/problem?id=2387

一个dijkstra的模板题

#include <stdio.h>
#include <string.h>
#define inf 999999;

int graph[ 1005 ][ 1005 ];
bool mark[ 1005 ];
int m,n;

void dijkstra()
{
    int d[ 1005 ],k,Min;
    bool mark[ 1005 ];
    for(int i = 1 ; i <= n ; i++)       //对d[i]进行初始化
    {
        mark[ i ] = false;
        d[ i ] = graph[ 1 ][ i ];
    }
    for(int i = 1 ; i <= n ; i++)   
    {
        Min=inf;
        for(int j = 1 ; j <= n ; j++)
        {
            if(!mark[ j ] && d[ j ] < Min)    
            {
                k = j;
                Min = d[ j ];
            }
        }
        mark[ k ] = true;

        for(int j = 1 ; j <= n ; j++)
        {
            if(!mark[ j ] && d[ j ] > graph[ k ][ j ] + d[ k ])
                d[ j ] = graph[ k ] [ j ] + d[ k ];
        }
    }
    printf("%d
",d[n]);
}

int main()
{
    int a,b,c;
    while(~scanf("%d%d",&m,&n)){
        for(int i = 0 ; i <= n ; i++)
            for(int j = 0 ; j <= n ; j++)
                if(i == j)
                    graph[ i ][ j ] = 0;
                else
                {
                    graph[ i ][ j ] = inf;
                    graph[ j ][ i ] = inf;
                }
            for(int i = 1 ; i <= m ;i++)
            {
                scanf("%d%d%d",&a,&b,&c);
                if(graph[ a ][ b ] > c)
                {
                   graph[ a ][ b ] = c;
                   graph[ b ][ a ] = c;
                }
            }
            dijkstra();
    }
    return 0;
}

dijkstra +head

const int N=405;
struct rec
{
    int v,w;//v代表所连的值，W代表权值
};


vector<rec> edge[N*N];
int n,st,ed;
long long dis[N*N];
bool vis[N*N];

struct cmp
{
    bool operator()(int a,int b)
    {
        return dis[a]>dis[b];
    }
};

void Dijkstra()
{
    priority_queue<int,vector<int>,cmp> Q;
    memset(dis,-1,sizeof(dis));
    memset(vis,0,sizeof(vis));
    int i,u,v;
    Q.push(st);
    dis[st]=1;
    while(!Q.empty())
    {
        u=Q.top();
        Q.pop();
        vis[u]=0;
        if(u==st)
            break;
            if(dis[v]==-1 || dis[v]<dis[u]*edge[u][i].w)
            {
                dis[v]=dis[u]*edge[u][i].w;
                if(!vis[v])
                {
                    vis[v]=1;
                    Q.push(v);
                }
            }
        }
    }
}
        for(i=0; i<edge[u].size(); i++)
        {
            v=edge[u][i].v;
            if(dis[v]==-1 || dis[v]<dis[u]*edge[u][i].w)
            {
                dis[v]=dis[u]*edge[u][i].w;
                if(!vis[v])
                {
                    vis[v]=1;
                    Q.push(v);
                }
            }
        }
    }
}