一道比较考思维的树形dp。
如果直接定义f[i][j]为i所在子树中取j个黑色的最大值的话是没办法转移的吧。。因此我们不妨考虑一条边带来的影响。如果令f[i][j]为i所在子树中取j个黑色时,i所在子树中边的影响总和最大的值,就很方便转移了。所谓边的影响,是指:
如果有一条边(x,y),显然这条边将一棵树分成了两块,那么如果一边有a个黑点,b个白点;另一边有c个黑点,d个白点,那么显然这条边对答案的影响(贡献)就是a*c+b*d了对吧。然后来看转移:
对于i的一个儿子k,考虑(i,k)这条边带来的影响,那么如果i中有j个黑色,k中有t个黑色,那么(i,j)这条边带来的影响就是(K-t)*t*len(i,j)+(sz[k]-t)*(N-K-(sz[k]-t))*len(i,j),sz为子树的大小。那么转化成动规方程式就是
f[i][j]=max{f[i][j-t]+f[k][t]+上面那个式子}。
这样显然f[1][K]就是答案了。
不过这样可能时间复杂度不太好算。注意到首先有K这个系数,然后每个点在父亲节点会有N这个系数,因此是O(NK)的。
AC代码如下(动规形式要改成刷表不然会很慢(←像我这样就很慢)):
#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
#define N 2005
using namespace std;
int n,m,tot,fst[N],pnt[N<<1],len[N<<1],nxt[N<<1],sz[N]; ll f[N][N];
void add(int x,int y,int z){
pnt[++tot]=y; len[tot]=z; nxt[tot]=fst[x]; fst[x]=tot;
}
void dp(int x,int fa){
int p,i,j; sz[x]=1;
for (i=2; i<=m; i++) f[x][i]=-1000000000;
for (p=fst[x]; p; p=nxt[p]){
int y=pnt[p];
if (y!=fa){
dp(y,x); sz[x]+=sz[y];
for (i=min(sz[x],m); i>=0; i--)
for (j=0; j<=sz[y] && j<=i; j++)
f[x][i]=max(f[x][i],f[x][i-j]+f[y][j]+(ll)len[p]*(j*(m-j)+(sz[y]-j)*(n-m-sz[y]+j)));
}
}
}
int main(){
scanf("%d%d",&n,&m); int i,x,y,z;
for (i=1; i<n; i++){
scanf("%d%d%d",&x,&y,&z);
add(x,y,z); add(y,x,z);
}
dp(1,0); printf("%lld\n",f[1][m]);
return 0;
}
by lych
2016.3.14