题解 | 还是畅通工程

//KY147 还是畅通工程
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int n, s[600];
long long sum=0;
struct E{
    int u, v, w;
    bool operator<(const E &x)const{
        return w<x.w;
    }
}e[6000];
void init(int n)
{
    for(int i=1;i<=n;i++) s[i]=i;
    sum=0;
    
}
int find(int x)
{
    if(x!=s[x]) s[x]=find(s[x]);
    return s[x];
}
void merge(int u, int v)
{
    u=find(u);
    v=find(v);
    if(u!=v) s[u]=v;
}
int main()
{
    while(cin>>n)
    {
        if(n==0) break;
        init(n);
        int cnt=0;
        for(int i=1;i<=n*(n-1)/2;i++){
            cin>>e[i].u>>e[i].v>>e[i].w;
        }
        sort(e+1,e+n*(n-1)/2+1);
        for(int i=1;i<=n*(n-1)/2;i++){
            if(cnt==n-1) break;
            int u=find(e[i].u);
            int v=find(e[i].v);
            if(u==v) continue;
            merge(u,v);
            sum+=e[i].w;
            cnt++;
        }
        cout<<sum<<'\n';
    }
    
}