题意:有一棵点数为 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;
}