码迷,mamicode.com
首页 > 编程语言 > 详细

图的最短路算法 Dijkstra及其优化

时间:2016-07-20 10:22:31      阅读:168      评论:0      收藏:0      [点我收藏+]

标签:

单源最短路径算法

时间复杂度O(N2) 优化后时间复杂度为O(MlogN)(M为图中的边数 所以对于稀疏图来说优化后更快)

不支持有负权的图

#include<iostream>
using namespace std;
const int maxn=1024;
const int inf=1<<30; 
int n,m;
int d[maxn];
int v[maxn]; 
int G[maxn][maxn];
void init()
{
	for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) G[i][j]=(i==j?0:inf);
	for(int i=2;i<=n;i++) d[i]=inf,v[i]=0;
	d[1]=0;//这里默认是以1作为起点的 
	v[1]=0;
}
int main()
{
	int from,to,dist;
	cin>>n>>m;
	init();
	for(int i=0;i<m;i++){
		cin>>from>>to>>dist;
		G[from][to]=G[to][from]=dist;
	}
	for(int i=1;i<=n;i++){
		int x,m=inf;
		for(int j=1;j<=n;j++){
			if(!v[j]&&d[j]<m)//写成‘<‘而不是‘<=‘必须要确保图是联通的
				m=d[x=j];
		} 
		v[x]=1;
		for(int j=1;j<=n;j++){
			if(G[x][j]<inf/*防止溢出*/&&G[x][j]+d[x]<d[j])
				d[j]=G[x][j]+d[x];
		}
	}
	return 0;
} 

  优化后代码

//迪杰斯特拉算法的优化
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
const int maxn=1024;
const int inf=1<<30; 
struct Edge{
	int f,t,d;
};
struct Node{
	int d,u;
	bool operator<(const Node& b)const{
		return d<b.d;
	} 
};
int n,m;
int d[maxn],v[maxn]; 
vector<int> G[maxn];
vector<Edge> edges;
priority_queue<Node> Q;
void init(){
	for(int i=1;i<=n;i++) G[i].clear();
	for(int i=1;i<=n;i++) v[i]=0;
	for(int i=2;i<=n;i++) d[i]=inf;
	d[1]=0; 
} 
int main()
{
	int from,to,dist; 
	cin>>n>>m;
	init();
	for(int i=0;i<m;i++){
		cin>>from>>to>>dist;
		edges.push_back((Edge){from,to,dist});
		G[from].push_back(edges.size()-1); 
		edges.push_back((Edge){to,from,dist});
		G[to].push_back(edges.size()-1);
	}
	Q.push((Node){0,1});
	while(!Q.empty()){
		Node x=Q.top();Q.pop();
		int u=x.u;
		if(v[u]) continue;
		v[u]=1;
		for(int i=0;i<G[u].size();i++){
			Edge& e=edges[G[u][i]];
			if(d[e.t]>d[u]+e.d){
				d[e.t]=d[u]+e.d;
				Q.push((Node){d[e.t],e.t}); 
			} 
		}		
	}
	return 0;
} 

  

图的最短路算法 Dijkstra及其优化

标签:

原文地址:http://www.cnblogs.com/gzhonghui/p/5686520.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!