【模板】堆優化 + dij +pair 存儲

 

 

就是短

 

感謝Cptraser dalao的博客

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 struct node {
 4     int val,num;
 5     bool operator <(const node &x) const {
 6         return val>x.val; 
 7     }
 8 };
 9 priority_queue <node> dij; 
10 vector < pair<int,int> > a[500001];
11 int n,m,s,dist[500001],done[500001]; //dist存儲邊權，done記錄是否已經進行了鬆弛操做；
12 int main(){
13     cin>>n>>m>>s;
14     for(int i=1;i<=m;i++){
15         int x,y,c;
16         cin>>x>>y>>c;
17         a[x].push_back(make_pair(y,c));
18     }
19     for(int i=1;i<=m;i++){
20         dist [i]=0x7fffffff;//便於進行鬆弛 luogu的弱化版有一個測試點就是卡的這個，七個f打全。
21     }
22     dist [s]=0;
23     dij.push((node){0,s});//放入堆頂元素
24     while (!dij.empty()){//dij標準格式233
25         int front=dij.top().num;//重複取出堆頂    若是已經拓展過就continue     鬆弛操做同時知足條件放入堆 這三個操做26         dij.pop();
27         if(done[front]) continue;
28         done[front]=true;
29         for(int i=0;i<a[front].size();i++){
30             int to=a[front][i].first,vl=a[front][i].second;
31             if(!done[to] && dist[front]+vl<dist[to]){
32                 dist[to]=dist[front]+vl;
33                 dij.push((node){dist[to],to});
34             }
35         }
36     }
37     for(int i=1;i<=n;i++) {
38         cout<<dist[i]<<" ";
39     }
40     return 0;
41 }

 

 

 

留坑之後寫pair的玄學用法