长春现场赛E题 zoj 3659

E：想法很巧妙的题目。
从大到小排序边加入，设每次边两边的连通块分别为x，y，边权为c，那么将x块与y块合并，x块中

点要增加的边权就是|y|×c，y块增加的边权就是|x|*c。可以用并查集实现合并，同时合并后只需

要将并查集的头设为当前块中边权和最大的点，每次只要维护一个点即可。

代码如下：

#include"stdio.h"
#include"string.h"
#include"stdlib.h"
#define maxn 200005

int fa[maxn],cnt[maxn];
long long total[maxn];

struct node
{
    int a,b,c;
}edge[maxn];

int cmp(const void *a,const void *b)
{
    node *aa=(node*)a,*bb=(node*)b;
    return bb->c-aa->c;
}

int find(int x)
{
    int temp=x;
    while(temp!=fa[temp])
        temp=fa[temp];
    fa[x]=temp;
    return temp;
}


int main( )
{
    int n,i,x,y,a,b,c;
    long long temp1,temp2;
    while(~scanf("%d",&n))
    {
        for(i=0;i<=n;i++)
        {
            fa[i]=i;
            cnt[i]=1;
            total[i]=0;
        }
        for(i=0;i<n-1;i++)
            scanf("%d%d%d",&edge[i].a,&edge[i].b,&edge[i].c);
        qsort(edge,n-1,sizeof(node),cmp);
        for(i=0;i<n-1;i++)
        {
            a=edge[i].a;b=edge[i].b;c=edge[i].c;
            x=find(a);y=find(b);
            temp1=(long long)cnt[y]*c+total[x];
            temp2=(long long)cnt[x]*c+total[y];
            if(temp1>temp2)
            {
                fa[y]=x;
                total[x]=temp1;
                cnt[x]+=cnt[y];
            }
            else
            {
                fa[x]=y;
                total[y]=temp2;
                cnt[y]+=cnt[x];
            }
        }
        printf("%lld\n",total[find(1)]);
    }
    return 0;
}