堆优化的dijkstra(迪杰斯特拉)

前面在讲prim算法的时候已经提到了dijkstra,这里再说说求最短路的dijkstra算法;

相比于邻接矩阵的存图方式我们可以用更适用于做题的链式前向星,更省空间;

我们也可以用STL的priority_queue优先队列进行优化(手写堆太麻烦了~~)

下面看代码(写的是求各个点到其他各个点的最短距离,可以用来当模板)

#include<cmath>
#include<queue>//优先队列是队列的一种
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=100010;
int to[maxn],m,bi[maxn],ne[maxn],ww[maxn],e,p[maxn],dis[maxn];
struct node{
    int hao,w;
};
bool operator <(node a,node b){
    return (a.w>b.w);//这样重载运算符写的是小根堆
}
struct dj{
    void init(){//初始化
        memset(p,0,sizeof(p));
        memset(dis,0,sizeof(dis));
        e=0;
    }
    void add(int x,int y,int z){//链式前向星存图
        to[++e]=y;
        ne[e]=bi[x];
        bi[x]=e;
        ww[e]=z;
    }
    void dijie(int h){
        for(int i=1;i<=m;i++)dis[i]=999999999;
        priority_queue<node>o;//定义一个node形的优先队列
        dis[h]=0;
        o.push((node){h,0});
        while(!o.empty()){
            node flag=o.top();
            o.pop();
            int u=flag.hao;
            if(p[u])continue;
            int v=flag.w;
            p[u]=1;
            dis[u]=v;
            for(int i=bi[u];i;i=ne[i]){
                int t=to[i];
                if(dis[t]>dis[u]+ww[i])
                dis[t]=dis[u]+ww[i];
                o.push((node){t,dis[t]});
            }
        }
    }
    void out(){
        for(int i=1;i<=m;i++)printf("%d ",dis[i]);
    }
}s;
int main(){
    int i,j,k,n,x,y,z;
    scanf("%d%d",&m,&n);
    for(i=1;i<=n;i++){
        scanf("%d%d%d",&x,&y,&z);
        s.add(x,y,z);
        s.add(y,x,z);
    }
    for(i=1;i<=m;i++){ 
    s.dijie(i);
    s.out();
    printf("\n");
    s.init();
}
 return 0;
}
    原文作者:Dijkstra算法
    原文地址: https://blog.csdn.net/ergedathouder/article/details/52439438
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞