堆優化的dijkstra(迪傑斯特拉)
阿新 • • 發佈:2018-12-24
前面在講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; }