bzoj4033 T1 树形dp

        一道比较考思维的树形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