HDU 4118 Holiday's Accommodation

题面

题意

给出一棵树，每个点有一个人，每条边有一个长度，现在每个人要到另外一个点，且不同人不能到相同的点，要求他们走的最短距离之和最大。

做法

如果直接考虑每个人走到哪一个点，是阶乘级别的，而且好像没有什么好的贪心方法。
但是可以从边对答案的贡献来考虑，一条边最多被经过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);
    }
}