diff --git a/TangDou/AcWing/MinimalSpanningTree/858.md b/TangDou/AcWing/MinimalSpanningTree/858.md index 35c0735..7c3ef57 100644 --- a/TangDou/AcWing/MinimalSpanningTree/858.md +++ b/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; @@ -203,53 +205,49 @@ const int N = 510; const int INF = 0x3f3f3f3f; int n, m; -int g[N][N]; //稠密图,邻接矩阵 -int dis[N]; //这个点到集合的距离 -bool st[N]; //是不是已经使用过 -int res; //最小生成树里面边的长度之和 -int pre[N]; //前驱结点 - -/** - * 功能:普利姆算法求最小生成树 - * @return - */ +int g[N][N]; // 稠密图,邻接矩阵 +int dis[N]; // 这个点到集合的距离 +bool st[N]; // 是不是已经使用过 +int res; // 最小生成树里面边的长度之和 +int pre[N]; // 前驱结点 + +// 普利姆算法求最小生成树 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)res += dis[t]; + 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]) { - dis[j] = g[t][j]; - pre[j] = t;//记录是由谁转移而来 - } + 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; cin >> a >> b >> c; 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]); return 0; }