ZOJ 3659Conquer a New Region解题报告

这道题用的是并查集，刚刚看到这道题的时候想起来自己OJ的一道有点相似的题，说是每条边有一个等级，两个点之间的连接等级为路径上所有边的最高等级，很像，但是因为这道题的边有权值而且还要算最大值，结果自己这个想法就只是想了想，然后看了别人的解题报告，还真是并查集，看来自己的第一反应还挺重要的，把边降序排序，这样保证新加的一条边肯定是最短的，分别对这条边连接的两个点作为这一部分中选中的点找最优值，直到把所有的边加完

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#define N 200005
#define max(a,b) ((a)>(b)?(a):(b))
using namespace std;
struct node
{
    long long u,v;
    long long w;
};
node e[N];
int cmp(const void *a,const void *b)
{
    node *c=(node *)a;
    node *d=(node *)b;
    if(d->w>c->w)
        return 1;
    else
        return -1;
}
long long f[N],num[N];
long long sum[N];
void init()
{
    int i;
    for(i=1;i<N;i++)
    {
        f[i]=i;
        sum[i]=0;
        num[i]=1;
    }
}
long long find(long long x)
{
    long long r=x;
    while(r!=f[r])
    {
        r=f[r];
    }
    long long t1;
    while(x!=f[x])
    {
        t1=f[x];
        f[x]=r;
        x=t1;
    }
    return r;
}
int main()
{
    long long n,i,j,k,fa,fb;
    long long ans;
    long long atob,btoa;
    while(scanf("%lld",&n)!=EOF)
    {
        init();
        for(i=0;i<n-1;i++)
        {
            scanf("%lld%lld%lld",&e[i].u,&e[i].v,&e[i].w);
        }
        qsort(e,n-1,sizeof(node),cmp);
        ans=0;
        for(i=0;i<n-1;i++)
        {
            fa=find(e[i].u);
            fb=find(e[i].v);
            atob=num[fa]*e[i].w+sum[fb];
            btoa=num[fb]*e[i].w+sum[fa];
            if(atob>btoa)
            {
                f[fa]=fb;
                sum[fb]=atob;
                num[fb]+=num[fa];
            }
            else
            {
                f[fb]=fa;
                sum[fa]=btoa;
                num[fa]+=num[fb];
            }
            ans=max(ans,max(atob,btoa));
        }
        printf("%lld\n",ans);
    }
    return 0;
}