day10

最小生成树

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;
}