'commit'

TangDou/AcWing/MinimalSpanningTree/858.md

@@ -161,7 +161,9 @@ int prim() {
         if (i) res += dis[t];
 
         // 4、因为本轮选择的是结点t,那么用t更新其它未加入到集合中点到集合的距离
-        for (int j = 1; j <= n; j++) dis[j] = min(dis[j], g[t][j]);
+        for (int j = 1; j <= n; j++)
+            if (!st[j] && dis[j] > g[t][j])
+                dis[j] = g[t][j];
 
         // 5、把t放到集合中
         st[t] = true;
@@ -209,36 +211,30 @@ bool st[N];     //是不是已经使用过
 int res;     // 最小生成树里面边的长度之和
 int pre[N];  // 前驱结点
 
-/**
- * 功能：普利姆算法求最小生成树
- * @return
- */
+// 普利姆算法求最小生成树
 int prim() {
-    //迭代n次
-    for (int i = 0; i < n; i++) {
+    for (int i = 0; i < n; i++) { // 迭代n次
         int t = -1;
         for (int j = 1; j <= n; j++)
             if (!st[j] && (t == -1 || dis[t] > dis[j])) t = j;
-        if (i && dis[t] == INF) return INF;
+        if (i && dis[t] == INF) return INF; // 非连通图，没有最小生成树
         if (i) res += dis[t];
         for (int j = 1; j <= n; j++)
-            if (!st[j]) {
-                if (g[t][j] < dis[j]) {
+            if (!st[j] && g[t][j] < dis[j]) {
                 dis[j] = g[t][j];
                 pre[j] = t; // 记录是由谁转移而来
             }
-            }
-        st[t] = 1;
+        st[t] = true;
     }
     return res;
 }
 
-
 int main() {
     cin >> n >> m;
     memset(g, 0x3f, sizeof g);
-    memset(dist, 0x3f, sizeof dist);
-    memset(pre, -1, sizeof pre);
+    memset(dis, 0x3f, sizeof dis);
+    memset(pre, -1, sizeof pre); // 记录前驱路径
+
     // 读入数据
     while (m--) {
         int a, b, c;
@@ -246,8 +242,10 @@ int main() {
         g[a][b] = g[b][a] = min(g[a][b], c);
     }
     int t = prim();
-    if (t == INF) puts("impossible");
-    else printf("%d\n", t);
+    if (t == INF)
+        puts("impossible");
+    else
+        cout << t << endl;
 
     // 输出前驱结点
     for (int i = 1; i <= n; i++) printf("%d ", pre[i]);