2021.01.31
最小生成树
Kruskal:
#include<bits/stdc++.h>
using namespace std;
const int maxn = 110;
struct Edge{
int u, v, w;
bool operator < (const Edge &r)const{ //边权从小到大排序
return w < r.w;
}
}E[maxn * maxn];
int n, m, f[maxn];
int find(int x) {
return x == f[x] ? x : f[x] = find(f[x]);
}
void Kruskal(int n, int m) {
int res = 0; // 存放结果
int num = 0; // 记录当前选择了多少条边
sort(E + 1, E + 1 + m);
for(int i = 1; i <= n; i++) f[i] = i; //初始化 每个点自己为一个集合
for(int i = 1; i <= m; i++) {
int f1 = find(E[i].u); // 查询u顶点在哪个集合中
int f2 = find(E[i].v); // 查询v顶点在哪个集合中
if(f1 != f2) { // 如果不在同一个集合中
num++; // 选中的边数 +1
res += E[i].w; // 答案加上这条边的权值
f[f1] = f2; // 将这两个点合并到一个集合中
}
if(num == n - 1) // 如果已经找到了 n - 1条边,说明最小生成树已经构建完成了
break;
}
if(num == n - 1) printf("%d\n", res);
else puts("?");
}
int main() {
while(~scanf("%d %d", &m, &n)) {
if(m == 0) break;
for(int i = 1; i <= m; i++)
scanf("%d%d%d", &E[i].u, &E[i].v, &E[i].w);
Kruskal(n, m);
}
return 0;
}
Prim
#include<bits/stdc++.h>
using namespace std;
const int maxn = 110;
#define inf 0x3f3f3f3f
int n, m, G[maxn][maxn], vis[maxn], d[maxn];
void Prim(int n, int m) {
memset(vis, 0, sizeof(vis));
int index = 1; // 当前加入到小树的顶点
int res = 0; // 存放结果
vis[index] = 1;
for(int i = 1; i <= n; i++) // 更新这个点到其他点的距离
d[i] = G[index][i];
for(int i = 1; i < n; i++) { // 执行 n - 1 次, 找剩下的n - 1 个点
int minn = inf;
for(int j = 1; j <= n; j++) { // 找出未加入小树且 d 最小的点
if(!vis[j] && d[j] < minn) {
minn = d[j];
index = j;
}
}
if(minn == inf) { // 如果没有找到, 说明不存在最小生成树
puts("?");
return ;
}
res += minn; // 累加答案
vis[index] = 1; // 将这个点加入最小生成树中
for(int j = 1; j <= n; j++) { // 更新这个点加入后,当前这棵小树到未加入的点的最近距离
if(!vis[j] && d[j] > G[index][j]) d[j] = G[index][j];
}
}
printf("%d\n", res);
}
int main() {
while(~scanf("%d%d", &m, &n)) {
if(m == 0) break;
for(int i = 1; i <= n; i++) { // 邻接矩阵的初始化
for(int j = 1; j <= n; j++) {
if(i == j) G[i][j] = 0;
else G[i][j] = inf;
}
}
for(int i = 1, u, v, w; i <= m; i++) {
scanf("%d%d%d", &u, &v, &w);
G[u][v] = G[v][u] = min(w, G[u][v]); // 消除重边的影响
}
Prim(n, m);
}
return 0;
}