题意:给出一棵树,每条边有一个容量,求一个中心点,让这个点到其他点的最大流量的和最大。
思路:开始还在想是不是树形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;
}