7-13（图） 畅通工程之局部最小花费问题（35 分） 35分代码

最小生成树

克鲁斯卡尔算法

首先要将已经修建的道路进行并查集合并操作

用 set 存集合中结点的个数，，剩余的点按照基本操作进行就好了

（注：博客作为交流使用，切勿抄袭应付作业）

#include<bits/stdc++.h>
using namespace std;
const int maxn = 100+7, INF = 0x7f7f7f7f;

int n, m;
int f[maxn];
set<int> st;

struct edge {
    int u, v, c, f;
}e[maxn*maxn];
bool cmp(edge a, edge b) {
    if(a.f == b.f) return a.c < b.c;
    return a.f > b.f;
}

void init() {
    scanf("%d", &n);
    m = n*(n-1)/2;
    for(int i = 0; i < m; ++i) {
        scanf("%d %d %d %d", &e[i].u, &e[i].v, &e[i].c, &e[i].f);
    }
    sort(e, e+m, cmp);
    for(int i = 0; i <= n; ++i)
        f[i] = i;
}

int find_(int x) {
    return f[x] == x ? x : f[x] = find_(f[x]);
}

void solve() {
    int ans = 0;
    for(int i = 0; i < m; ++i) {
        int uu = find_(e[i].u), vv = find_(e[i].v);
        st.insert(e[i].u); st.insert(e[i].v);
        if(e[i].f == 1) {
            f[uu] = vv;
        }
        else if(uu != vv){
            ans += e[i].c;
            f[uu] = vv;
        }
        if(st.size() == n) break;
    }
    cout << ans << endl;
}

int main() {
    init();
    solve();
    return 0;
}

/*
4
1 2 1 1
1 3 4 0
1 4 1 1
2 3 3 0
2 4 2 1
3 4 5 0
*/