NC20131 [JLOI2011]飞行路线

https://ac.nowcoder.com/acm/contest/view-submission?submissionId=52418973&returnHomeType=1&uid=742234351

NowCoder/20131/20131.cpp

@@ -0,0 +1,77 @@
+#include <algorithm>
+#include <cstring>
+#include <iostream>
+#include <queue>
+#include <vector>
+
+using std::cin;
+using std::cout;
+const char endl = '\n';
+
+const int N = 10005;
+
+int n, m, k, s, t;
+std::vector<std::pair<int, int>> g[N << 4];
+
+int dist[N << 4];
+bool vis[N << 4];
+
+void dijkstra() {
+    memset(dist, 0x3f, sizeof(dist));
+    memset(vis, 0x00, sizeof(vis));
+
+    std::priority_queue<std::pair<int, int>, std::vector<std::pair<int, int>>, std::greater<std::pair<int, int>>> q;
+
+    q.push(std::make_pair(0, s));
+    dist[s] = 0;
+
+    while (!q.empty()) {
+        auto t = q.top();
+        q.pop();
+
+        if (vis[t.second]) continue;
+
+        for (auto e : g[t.second]) {
+            int v = e.first,
+                w = e.second;
+
+            if (dist[v] > t.first + w) {
+                dist[v] = t.first + w;
+
+                q.push(std::make_pair(dist[v], v));
+            }
+        }
+
+        vis[t.second] = true;
+    }
+}
+
+int main() {
+    std::ios::sync_with_stdio(false);
+
+    cin >> n >> m >> k >> s >> t;
+
+    for (int i = 1, u, v, w; i <= m; i++) {
+        cin >> u >> v >> w;
+
+        g[u].push_back(std::make_pair(v, w));
+        g[v].push_back(std::make_pair(u, w));
+
+        for (int j = 1; j <= k; j++) {
+            g[u + (j - 1) * n].push_back(std::make_pair(v + j * n, 0));
+            g[v + (j - 1) * n].push_back(std::make_pair(u + j * n, 0));
+            g[u + j * n].push_back(std::make_pair(v + j * n, w));
+            g[v + j * n].push_back(std::make_pair(u + j * n, w));
+        }
+    }
+
+    for (int i = 1; i <= k; i++) {
+        g[t + (i - 1) * n].push_back(std::make_pair(t + i * n, 0));
+    }
+
+    dijkstra();
+
+    cout << dist[t + n * k] << endl;
+
+    return 0;
+}