Dijkstra最短路徑算法的優化

在傳統的Dijkstra算法中，我們不難發現，大量的時間被用於遍歷d[]數組。
因此，我們可以通過一個小頂堆來替代這個遍歷過程。優化後的算法複雜度爲O（v*lgn）。

#include <cstdio>
#include <cstring>
#include <queue>
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<Pair>,greater<Pair> > 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;
}