'commit'

TangDou/AcWing_TiGao/T3/MiniPathExtend/1137_1.cpp

@@ -0,0 +1,62 @@
+#include <bits/stdc++.h>
+using namespace std;
+typedef pair<int, int> PII;
+// 建立虚拟源点0
+const int N = 1010, M = 40010;
+const int INF = 0x3f3f3f3f;
+int n, m, S;
+
+// 邻接表
+int h[N], e[M], w[M], ne[M], idx;
+void add(int a, int b, int c) {
+    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
+}
+int dis[N]; // 最短距离数组
+bool st[N]; // 是否进过队列
+// 迪杰斯特拉
+void dijkstra() {
+    memset(dis, 0x3f, sizeof dis);                    // 初始化大
+    memset(st, 0, sizeof st);                         // 初始化为未出队列过
+    priority_queue<PII, vector<PII>, greater<PII>> q; // 小顶堆
+    q.push({0, 0});                                   // 出发点入队列
+    dis[0] = 0;                                       // 出发点距离0
+
+    while (q.size()) {
+        auto t = q.top();
+        q.pop();
+        int u = t.second;
+        if (st[u]) continue;
+        st[u] = true;
+        for (int i = h[u]; ~i; i = ne[i]) {
+            int j = e[i];
+            if (dis[j] > dis[u] + w[i]) {
+                dis[j] = dis[u] + w[i];
+                q.push({dis[j], j});
+            }
+        }
+    }
+    // 注意：此处的S是终点，不是起点，不是起点，不是起点！
+    printf("%d\n", dis[S] == INF ? -1 : dis[S]);
+}
+int main() {
+    while (cin >> n >> m >> S) {
+        // 注意清空链表头
+        memset(h, -1, sizeof h);
+        idx = 0;
+        // m条边
+        while (m--) {
+            int a, b, c;
+            cin >> a >> b >> c;
+            add(a, b, c);
+        }
+        int T;
+        cin >> T;
+        while (T--) {
+            int x;
+            cin >> x;
+            add(0, x, 0); // 超级源点法
+        }
+        dijkstra();
+    }
+    return 0;
+}