关闭

【bzoj4033】 T1 树形dp

249人阅读 评论(0) 收藏 举报
分类:

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

    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 2010

using 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

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:146034次
    • 积分:4408
    • 等级:
    • 排名:第7041名
    • 原创:302篇
    • 转载:0篇
    • 译文:0篇
    • 评论:69条
    最新评论