Dijkstra+堆優(yōu)化

#include<iostream>
#include<cstdio>

using namespace std;

int n,m,u,v,w,q;
int head[10001],next[200001],data[200001],value[200001];
int heap[10001],dis[10001];

void update(){
    int x,y;
    x=heap[1];
    y=head[x];
    while (y!=0){
        dis[data[y]]=min(dis[x]+value[y],dis[data[y]]);
        y=next[y];
    }
}

void adjust(int x){
    int p,minDis;
    p=x;
    minDis=dis[heap[x]];
    if (x*2<=n)
        if (dis[heap[x*2]]<minDis){
            p=x*2;
            minDis=dis[heap[x*2]];
        }
    if (x*2+1<=n)
        if (dis[heap[x*2+1]]<minDis){
            p=x*2+1;
            minDis=dis[heap[x*2+1]];
        }
    if (p!=x){
        swap(heap[x],heap[p]);
        adjust(p);
    }
}

void dijkstra(int x){
    int i,s=n;
    for (i=1;i<=n;i++)
        heap[i]=i;
    dis[1]=0;
    for (i=2;i<=n;i++)
        dis[i]=1111111111;
    while (s>0){
        update();
        swap(heap[1],heap[s]);
        s--;
        for (i=s/2;i>=1;i--)
            adjust(i);
    }
}

int main(){
    int i,j;
    scanf("%d%d",&n,&m);
    for (i=1;i<=m;i++){
        scanf("%d%d%d",&u,&v,&w);
        next[i*2-1]=head[u],data[i*2-1]=v,value[i*2-1]=w,head[u]=i*2-1;
        next[i*2]=head[v],data[i*2]=u,value[i*2]=w,head[v]=i*2;
    }
    
    dijkstra(1);
    return 0;
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容