BZOJ 4033: [HAOI2015]T1

题意：有一棵点数为 N 的树，树边有边权。给你一个在 0~ N 之内的正整
数 K ，你要在这棵树中选择 K个点，将其染成黑色，并将其他 的
N-K个点染成白色 。 将所有点染色后，你会获得黑点两两之间的距
离加上白点两两之间的距离的和的受益。问受益最大值是多少。

为什么看到这种题会想到点分治…其实树规就好了，记f[i][j]表示以i为根的子树取了j个黑点的收益，向祖先转移的时候只要考虑从自己到祖先那条边的贡献，记sz[i]为以i为根的子树大小,则贡献为j*(k-j)c+(sz[i]-j)(n-k-sz[i]+j)*c（c为边权），接下来就是一个树形背包啦，因为有些状态不能到达，所以转移的时候可以由合法状态顺着转移，而不是倒着逆推（反正我写挂了…)

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
using namespace std;
const int maxn=2000+10;
long long f[maxn][maxn],tmp[maxn];
int n,K,sz[maxn];
struct node
{
  int to,cost;
};
vector<node> g[maxn];
void dfs(int p,int fa,int w)
{
  sz[p]=1;
  for(int i=0;i<g[p].size();i++)
  {
    int v=g[p][i].to,c=g[p][i].cost;if(v==fa) continue;
    dfs(v,p,c);
    memcpy(tmp,f[p],sizeof(f[p]));
    for(int j=0;j<=min(sz[v],K);j++)
      for(int k=0;k<=min(sz[p],K-j);k++)
        tmp[j+k]=max(tmp[j+k],f[v][j]+f[p][k]);
    sz[p]+=sz[v];
    memcpy(f[p],tmp,sizeof(tmp));
  }
  for(int i=0;i<=min(sz[p],K);i++)
    f[p][i]+=(long long)i*(long long)(K-i)*(long long)w+(long long)(n-K-(sz[p]-i))*(long long)(sz[p]-i)*(long long)w;
}
int main()
{
  //freopen("4033.in","r",stdin);
  //freopen("4033.out","w",stdout);
  scanf("%d%d",&n,&K);
  for(int i=1;i<n;i++)
  {
    int x,y,c;scanf("%d%d%d",&x,&y,&c);
    g[x].push_back((node){y,c});
    g[y].push_back((node){x,c});
  }
  dfs(1,-1,0);
  printf("%lld\n",f[1][K]);
  return 0;
}