-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdijkstra.cpp
86 lines (74 loc) · 2.62 KB
/
dijkstra.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
/*
AUTHOR : Peeyush Yadav
Problem : dijkstra's algorithm
*/
#include<bits/stdc++.h>
using namespace std; typedef long long ll; typedef pair<int,int> grp;
inline ll power(ll a,ll b) { ll r=1; while(b){ if(b&1) r=r*a ; a=a*a ; b>>=1;} return r;}
inline ll power(ll a,ll b,ll m){ ll r=1; while(b){ if(b&1) r=(r*a)%m; a=(a*a)%m; b>>=1;} return r;}
void fast(){
#ifdef Megamind
freopen("inp.txt","r",stdin);
//freopen("out.txt","w",stdout);
#define trace(x) cerr<<__FUNCTION__<<":"<<__LINE__<<": "#x" = "<<x<<endl;
#define trace2(x,y) cerr<<__FUNCTION__<<":"<<__LINE__<<": "#x" = "<<x<<" | "#y" = "<<y<<endl;
#define trace3(x,y,z) cerr<<__FUNCTION__<<":"<<__LINE__<<": "#x" = "<<x<<" | "#y" = "<<y<<" | "#z" = "<<z<<endl;
#else
#define trace(x)
#define trace2(x,y)
#define trace3(x,y,z)
#endif
#define fi first
#define se second
#define mk make_pair
#define pb(x) push_back(x)
#define s(x) scanf("%d",&x);
#define sl(x) scanf("%lld",&x);
#define p(x) printf("%d\n",x);
#define f(a,b,c) for(int a=b;a<c;a++)
#define r(a,b,c) for(int a=b;a>c;a--)
#define p2(x,y) printf("%d %d\n",x,y);
#define pl(x) printf("%lld\n",x);
#define pl2(x,y) printf("%lld %lld\n",x,y);
#define p1d(a,n) for(int ix=0;ix<n;ix++) printf("%d ",a[ix]); printf("\n");
#define p2d(a,n,m) for(int ix=0;ix<n;ix++){ for(int jx=0;jx<m;jx++) printf("%d ",a[ix][jx]); printf("\n");}
}
/*........................................................END OF TEMPLATES.......................................................................*/
#define sz 112345
vector<grp> v[sz];
vector<int> dist(sz, 987654321);
void dijkstra(int s, int dest){
priority_queue<grp,vector<grp>, greater<grp> > Q;
dist[s] = 0;
Q.push(grp(0,s));
while(!Q.empty()) {
grp top = Q.top();
Q.pop();
int parent = top.se, d = top.fi;
if(d <= dist[parent]) {
for(vector<grp> :: iterator it = v[parent].begin(); it != v[parent].end(); it++){
int child = it->fi, cost = it->se;
if(dist[child] > dist[parent] + cost) {
dist[child] = dist[parent] + cost;
Q.push(grp(dist[child],child));
}
}
}
}
}
int main(){
fast();
int node, edges, a, b, c;
s(node)
s(edges)
f(i,0,edges){
s(a) s(b) s(c)
v[a].pb(grp(b,c));
v[b].pb(grp(a,c));
}
dijkstra(0, 0);
f(i,0,node) trace(dist[i])
#ifdef Megamind
cerr << "\nTime elapsed : " << 1.0 * clock() / CLOCKS_PER_SEC * 1000 << " ms." << endl;
#endif
}