【bzoj4033】 T1 树形dp

来源：互联网发布：芜湖生活频道网络直播编辑：程序博客网时间：2024/05/24 03:23

表示长这么大第一次写树形背包，感觉有些难理解。

f[i][j]表示考虑以i为根这棵子树，有j个黑色结点的最大贡献值为多少。

则每次转移的时候我们需要考虑每条边对答案作出的贡献，为((k-j)*j+(size[i]-j)*(n-k+j-size[i]))*w。

利用那种类似树形背包的dp，O（n^2）解决。

#include<cstdio>#include<cstring>#include<cstdlib>#include<cmath>#include<iostream>#include<algorithm>#define maxn 2010using namespace std;int n,m,k,num;int head[maxn],to[2*maxn],next[2*maxn],v[2*maxn],size[maxn];long long tmp[maxn];long long f[maxn][maxn];void addedge(int x,int y,int z){num++;to[num]=y;v[num]=z;next[num]=head[x];head[x]=num;}long long calc(int x,int y){return (long long)y*(x-y);}void dfs(int x,int fa,int w){size[x]=1;for (int p=head[x];p;p=next[p])  if (to[p]!=fa)  {  dfs(to[p],x,v[p]);  memcpy(tmp,f[x],sizeof(tmp));  for (int i=0;i<=min(size[x],k);i++)    for (int j=0;j<=min(size[to[p]],k-i);j++)      tmp[i+j]=max(tmp[i+j],f[x][i]+f[to[p]][j]);  memcpy(f[x],tmp,sizeof(tmp));  size[x]+=size[to[p]];  }for (int i=0;i<=min(size[x],k);i++)  f[x][i]+=(calc(k,i)+calc(n-k,size[x]-i))*w;}int main(){scanf("%d%d",&n,&k);for (int i=1;i<n;i++){int x,y,z;scanf("%d%d%d",&x,&y,&z);addedge(x,y,z);addedge(y,x,z);}dfs(1,0,0);printf("%lld\n",f[1][k]);return 0;}

0 0