1. 程式人生 > >堆優化的dijkstra(迪傑斯特拉)

堆優化的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;
}