题面
题意
给出一棵树,每个点有一个人,每条边有一个长度,现在每个人要到另外一个点,且不同人不能到相同的点,要求他们走的最短距离之和最大。
做法
如果直接考虑每个人走到哪一个点,是阶乘级别的,而且好像没有什么好的贪心方法。
但是可以从边对答案的贡献来考虑,一条边最多被经过2*min(与它相连的两棵子树的大小)次,把所有边的最大贡献加起来即为答案。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
#define N 100100
using namespace std;
ll T,TT,n,first[N],ans,size[N],bb;
struct Bn
{
ll to,next,quan;
} bn[N<<1];
inline void add(int u,int v,ll w)
{
bb++;
bn[bb].to=v;
bn[bb].next=first[u];
bn[bb].quan=w;
first[u]=bb;
}
void dfs(ll now,ll last)
{
ll p,q;
size[now]=1;
for(p=first[now]; p!=-1; p=bn[p].next)
{
if(bn[p].to==last) continue;
dfs(bn[p].to,now);
size[now]+=size[bn[p].to];
ans+=min(size[bn[p].to],n-size[bn[p].to])*bn[p].quan;
}
}
int main()
{
ll i,j,p,q,o;
cin>>T;
while(T--)
{
ans=bb=0;
memset(first,-1,sizeof(first));
printf("Case #%lld: ",++TT);
scanf("%lld",&n);
for(i=1; i<n; i++)
{
scanf("%lld%lld%lld",&p,&q,&o);
add(p,q,o),add(q,p,o);
}
dfs(1,-1);
printf("%lld\n",ans*2);
}
}