POJ 1155 TELE 树形DP(背包)

点击打开链接

题意:有n个结点的树,m个叶结点代表用户的价值,其余边表示代价,问收入不为负数时,最多能选多少个叶结点? n<=3e3

设dp[i][k] 表示结点i选择k个叶结点的最大收入,ans：第一个dp[1][ans]>=0

转移状态时 枚举从子树中选择多少个结点,即分组背包 O(n^3)水过....

#include <cstdio>
#include <vector>
#include <algorithm>
#include <cstring>
#include <iostream>
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;
const int inf=1e7;
const int N=3e3+20;
int n,m,tot;
int a[N],sz[N];
int dp[N][N];
struct node{
	int to,w,next;
}e[N*2];
int head[N];
void addedge(int u,int v,int w)
{
	e[tot].to=v,e[tot].w=w;
	e[tot].next=head[u];
	head[u]=tot++;
}
void dfs(int u,int fa)
{
	if(u>n-m)//leaf
	{
		dp[u][1]=a[u];
		sz[u]=1;
		return;
	}
	for(int i=head[u];i!=-1;i=e[i].next)
	{
		int v=e[i].to,w=e[i].w;
		if(v!=fa)
		{
			dfs(v,u);
			sz[u]+=sz[v];
		}
	}
	for(int i=head[u];i!=-1;i=e[i].next)
	{
		int v=e[i].to,w=e[i].w;
		if(v!=fa)
		{
			dfs(v,u);
			for(int j=sz[u];j>=1;j--)
			{
				///从子树v中选k个,前几组子树中选j-k个
				for(int k=1;k<=j;k++) 
				{
					dp[u][j]=max(dp[u][j],w+dp[v][k]+dp[u][j-k]);	
				}
			}
		}
	}
}
int main()
{
	while(cin>>n>>m)
	{
		int k,u,c;
		tot=0;
		memset(head,-1,sizeof(head)); 
		memset(sz,0,sizeof(sz)); 
		for(int i=1;i<=n;i++)
		{
			a[i]=0;
			dp[i][0]=0;
			for(int j=1;j<=n;j++)
				dp[i][j]=-inf;
		}
		
		int cnt=0;
		for(int i=1;i<=n-m;i++)
		{
			scanf("%d",&k);
			for(int j=0;j<k;j++)
			{
				scanf("%d%d",&u,&c);
				addedge(i,u,-c);
			}
		}
		for(int i=n-m+1;i<=n;i++)
			scanf("%d",&a[i]);
		dfs(1,-1);
		for(int k=m;k>=0;k--)
		{
			if(dp[1][k]>=0)
			{
				printf("%d\n",k);
				break;
			}
		}
	}
	return 0;
}