Dijkstra最短路径算法的优化

在传统的Dijkstra算法中,我们不难发现,大量的时间被用于遍历d[]数组。
因此,我们可以通过一个小顶堆来替代这个遍历过程。优化后的算法复杂度为O(v*lgn)。

#include 
#include 
#include 
using namespace std;
#define MAXN 10005
#define MAXM 500005
#define LIMI 0x3f3f3f3f
#define MAXINT 0x7fffffff
struct Node{int s,e,w,n;}nd[MAXM];
typedef pair<int,int> Pair;
int bus[MAXN]={},m,n,cnt=0,ps,t1,t2,t3;
void AddEdge(int s,int e,int w)
{
    nd[++cnt].s=s; nd[cnt].e=e; nd[cnt].w=w;
    nd[cnt].n=bus[s];bus[s]=cnt;
    return;
}
void Dijkstra(int v0)
{
    bool u[MAXN];memset(u,0x00,sizeof u);
    int d[MAXN];memset(d,0x3f,sizeof d);
    priority_queue < Pair,vector,greater > q;
    q.push(make_pair(0,v0));d[v0]=0;
    while(!q.empty())
    {
        Pair p=q.top();q.pop();
        if(u[p.second]) continue;
        u[p.second]=true;
        for(int i=bus[p.second];i;i=nd[i].n)
            if(d[nd[i].e]>d[p.second]+nd[i].w){
                d[nd[i].e]=d[p.second]+nd[i].w;
                if(!u[nd[i].e]) q.push(make_pair(d[nd[i].e],nd[i].e));
            }
    }
    for(int i=1;i<=n;i++) printf("%d ",d[i]>=LIMI?MAXINT:d[i]);
    return;
}
int main()
{
    scanf("%d%d%d",&n,&m,&ps);
    for(int i=1;i<=m;i++) scanf("%d%d%d",&t1,&t2,&t3), AddEdge(t1,t2,t3);
    Dijkstra(ps);
    return 0;
}

你可能感兴趣的:(基础数据结构,图论)