Codeforces Round #536 (Div. 2) - D. Lunar New Year and a Wander（最短路）

  1 #include <bits/stdc++.h>
  2 
  3 using namespace std;
  4 
  5 #define REP(i, n) for (int i = 1; i <= (n); i++)
  6 #define sqr(x) ((x) * (x))
  7 
  8 const int maxn = 100000 + 100;
  9 const int maxm = 200000 + 100;
 10 const int maxs = 256;
 11 
 12 typedef long long LL;
 13 typedef pair<int, int> pii;
 14 typedef pair<double, double> pdd;
 15 
 16 const LL unit = 1LL;
 17 const int INF = 0x3f3f3f3f;
 18 const double eps = 1e-14;
 19 const double inf = 1e15;
 20 const double pi = acos(-1.0);
 21 const int SIZE = 100 + 5;
 22 const LL MOD = 1000000007;
 23 
 24 struct Edge
 25 {
 26     int to, next;
 27 } edge[maxm<<1];
 28 
 29 struct HeapNode
 30 {
 31     int u;
 32     bool operator< (const HeapNode &a)const
 33     {
 34         return u > a.u;
 35     }
 36 };
 37 
 38 int n, m;
 39 int head[maxn], tot;
 40 int dist[maxn];
 41 vector<int> ans;
 42 bool vis[maxn];
 43 
 44 void init()
 45 {
 46     memset(head, -1, sizeof(head));
 47     tot = 0;
 48 }
 49 
 50 void AddEdge(int u, int v)
 51 {
 52     edge[tot].to = v;
 53     edge[tot].next = head[u];
 54     head[u] = tot++;
 55 }
 56 
 57 void Dijkstra()
 58 {
 59     memset(dist, INF, sizeof(dist));
 60     dist[0] = 0;
 61     priority_queue<HeapNode> que;
 62     que.push((HeapNode{0}));
 63     while(!que.empty())
 64     {
 65         HeapNode x = que.top();
 66         int u = x.u;
 67         que.pop();
 68         vis[u] = true;
 69         ans.push_back(u + 1);
 70         for(int i = head[u]; i != -1; i = edge[i].next)
 71         {
 72             int v = edge[i].to;
 73             if(vis[v])
 74                 continue;
 75             if(dist[v] == INF)
 76             {
 77                 dist[v] = v;
 78                 que.push((HeapNode){v});
 79             }
 80         }
 81     }
 82 }
 83 
 84 int main()
 85 {
 86     ios::sync_with_stdio(false);
 87     cin.tie(0);
 88     //freopen("input.txt", "r", stdin);
 89     //freopen("output.txt", "w", stdout);
 90     cin >> n >> m;
 91     init();
 92     int u, v;
 93     for(int i = 0; i < m; i++)
 94     {
 95         cin >> u >> v;
 96         u--, v--;
 97         AddEdge(u, v);
 98         AddEdge(v, u);
 99     }
100     Dijkstra();
101     for(int i = 0; i < ans.size(); i++)
102     {
103         cout << ans[i] << " ";
104     }
105     return 0;
106 }