码迷,mamicode.com
首页 > 其他好文 > 详细

dijkstra

时间:2019-10-24 23:13:07      阅读:89      评论:0      收藏:0      [点我收藏+]

标签:return   jks   结束   pop   inline   max   pre   color   重载运算符   

这是一个用来求没有负边权的最短路径算法,复杂度是n^3,经过优先队列优化则是n^2.

算法思想:首先用前向星存储图,用一个node(需要重载运算符)类的priority_queue来存储被松弛的点(vis[i]==0)的的信息,dis[]数组存放当前到达这个点的最短路。其次进行扫描,看堆顶,也就是当前dis[]最小的点,先将其出队,再看其所连的点是否可以进行松弛操作,如果可以松弛,那么更新,并加入优先队列,当队列为空的时候结束。

代码

 1 #include<bits/stdc++.h>
 2 #define maxn  1000005
 3 #define maxm 2100000
 4 using namespace std;
 5 inline int read()
 6 {
 7     int x=0,k=1; char c=getchar();
 8     while(c<0||c>9){if(c==-)k=-1;c=getchar();}
 9     while(c>=0&&c<=9)x=(x<<3)+(x<<1)+(c^48),c=getchar();
10     return x*k;
11 }
12 struct edge{
13     int next;
14     int w;
15     int v;
16 }e[maxm];
17 int n,m,tot=0,s;
18 int head[maxn],vis[maxn],dis[maxn];
19 struct node{
20     int w,now;
21     inline bool operator <(const node &x)const
22     {
23         return w>x.w;//这里注意符号要为‘>‘
24     } 
25 };
26 inline void add(int u,int v,int w){
27     e[++tot].next=head[u];
28     head[u]=tot;
29     e[tot].w=w;
30     e[tot].v=v;    
31 }
32 priority_queue<node>q;
33 void dijkstra(){
34     for(int i=1;i<=n;i++) {
35         dis[i]=1234567890;
36     }
37     dis[s]=0;
38     q.push((node){0,s});
39     while(!q.empty()){
40         node x=q.top();
41         int u=x.now;        
42         q.pop();
43         if(vis[u]==1) continue;
44         vis[u]=1;
45         for(int i=head[u];i;i=e[i].next){
46             int v=e[i].v;
47             if(dis[v]>dis[u]+e[i].w){
48                 dis[v]=dis[u]+e[i].w;
49                 q.push((node){dis[v],v});
50             }
51         }
52     }
53 }
54 int main(){
55     n=read(),m=read(),s=read();
56     for(int i=1,x,y,z;i<=m;i++){
57         cin>>x>>y>>z;
58         add(x,y,z);
59     }
60     dijkstra();
61     for(int i=1;i<=n;i++){
62         cout<<dis[i]<<" ";
63     }
64     return 0;
65 }

 

dijkstra

标签:return   jks   结束   pop   inline   max   pre   color   重载运算符   

原文地址:https://www.cnblogs.com/china-mjr/p/11735516.html

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