标签:
单源最短路径算法
时间复杂度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; }
标签:
原文地址:http://www.cnblogs.com/gzhonghui/p/5686520.html