ZOJ 3659 Conquer a New Region(并查集)

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4882

题意：给出一棵树，边有权值。两个节点i和j之间的距离D(i,j)定义为i和j之间的路径上最小的权值。求一个点使得该点到其他点的D值和最大？

思路：并查集：将权值按照降序插入并查集。。。一开始我以为是树形DP呢，搞了半天。。

 

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define i64 long long
using namespace std;

struct node
{
    int size,p;
    i64 sum;
};

struct Node
{
    int u,v;
    i64 c;
};

Node b[200005];
node a[200005];
int n;

int cmp(Node a,Node b)
{
    return a.c>b.c;
}

int get(int u)
{
    if(a[u].p!=u) a[u].p=get(a[u].p);
    return a[u].p;
}

void deal()
{
    int i,u,v,ru,rv;
    i64 c,x,y;
    for(i=1;i<=n-1;i++)
    {
        u=b[i].u;
        v=b[i].v;
        c=b[i].c;
        ru=get(u);
        rv=get(v);
        x=a[ru].sum+c*a[rv].size;
        y=a[rv].sum+c*a[ru].size;
        if(x<y)
        {
            a[ru].p=rv;
            a[rv].size+=a[ru].size;
            a[rv].sum=y;
        }
        else
        {
            a[rv].p=ru;
            a[ru].size+=a[rv].size;
            a[ru].sum=x;
        }
    }
    printf("%lld\n",a[get(1)].sum);
}

int main()
{
    while(scanf("%d",&n)!=-1)
    {
        int i;
        for(i=1;i<n;i++) scanf("%d%d%lld",&b[i].u,&b[i].v,&b[i].c);
        sort(b+1,b+n,cmp);
        for(i=1;i<=n;i++)
        {
            a[i].size=1;
            a[i].sum=0;
            a[i].p=i;
        }
        deal();
    }
    return 0;
}