hdu 4424 Conquer a New Region(并查集)

题意：给出一棵树，每条边有一个容量，求一个中心点，让这个点到其他点的最大流量的和最大。

思路：开始还在想是不是树形dp，唉，我还是太弱了……首先把边排个序，然后一条边一条边合并，假如我们已经算出了要添加的边所连的两个子树的最大流量和，由于边是从大到小添加的，当前边的容量一定是一个树中的点到另一个树的最大流量，那么两个子树合并后的最大流量和就是某一个子树的最大流量和+另一个子树中的点的个数×当前边的容量，将算出的两个值取最大的就是合并后的值了。

 

代码：

 

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int maxn=200000+10;
struct Edge
{
    int u,v;
    ll w;
    bool operator < (const Edge &e )const
    {
        return w>e.w;
    }
};
Edge edges[maxn];
int cnt[maxn],parents[maxn];
ll sum[maxn];
int Find(int x)
{
    return x==parents[x]?x:parents[x]=Find(parents[x]);
}
void Uion(int x,int y,ll v)
{
    int a=Find(x);
    int b=Find(y);
    ll ta=sum[a]+cnt[b]*v;
    ll tb=sum[b]+cnt[a]*v;
    if(ta>tb)
    {
        sum[a]=ta;
        cnt[a]+=cnt[b];
        parents[b]=a;
    }
    else
    {
        sum[b]=tb;
        cnt[b]+=cnt[a];
        parents[a]=b;
    }
}
void Init(int n)
{
    for(int i=1;i<=n;++i)
    {
        parents[i]=i;
        sum[i]=0;
        cnt[i]=1;
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        int i;
        Init(n);
        for(i=0;i<n-1;++i)
          scanf("%d%d%I64d",&edges[i].u,&edges[i].v,&edges[i].w);
        sort(edges,edges+n-1);
        for(i=0;i<n-1;++i)
        {
            Uion(edges[i].u,edges[i].v,edges[i].w);
        }
        printf("%I64d\n",sum[Find(1)]);
    }
    return 0;
}