并查集+贪心 Conquer a New Region HDU 4118 && ZOJ 3659

据说是区域赛的题，拿到的时候可把我吓坏了，没想到是水水的一题

题意：

输入N表示N个城市，再输入N-1条路a,b,c，c是这条路的运载能力

计算以一个一个城市开始到各个城市运输的最大量

输入

5

1 2 1

2 3 1

2 4 1

4 5 1

则以2或4为中心，输出1+1+2+1=5

输入

5

1 2 1

2 3 1

2 4 1

4 5 1

则以4为中心，输出1+1+2=4

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define maxn 200005
using namespace std;
struct xxx
{
int u,v,w;
}num[maxn];            //存下输入的路，方便排序
int f[maxn];           //父节点
int level[maxn];       //当前点内包含点的个数
long long sum[maxn];   //当前点的运输量的和
int find(int i)        //找根节点
{
if(i==f[i])
return f[i];
f[i]=find(f[i]);
return f[i];
}
int cmp(xxx a,xxx b)
{
return a.w>b.w;
}
int main()
{
int n;
int i,j;
while(scanf("%d",&n)!=EOF)
{
for(int i=0;i<n-1;i++)
scanf("%d%d%d",&num[i].u,&num[i].v,&num[i].w);
sort(num,num+n-1,cmp);             //先排序，再处理
for(int i=1;i<=n;i++)              //初始化
{
sum[i]=0;
f[i]=i;
level[i]=1;
}
long long ans=0;
i=0;
for(i=0;i<n;i++)
{
int fr,ed;
fr=find(num[i].u);
ed=find(num[i].v);
if(fr!=ed)
{
long long t1,t2;                               //注意开long long，没开WA了一发
t1=sum[fr]+(long long )num[i].w*level[ed];    //比较两种方式较大的一种，选择合适的中心
t2=sum[ed]+(long long )num[i].w*level[fr];
if(t1>t2)
{
f[ed]=fr;
level[fr]+=level[ed];
sum[fr]=t1;
ans=t1;
}
else
{
f[fr]=ed;
level[ed]+=level[fr];
sum[ed]=t2;
ans=t2;
}
}
}
printf("%lld\n",ans);
}
return 0;
}