POJ 1155 TELE (树形DP+背包)

思路：设dp[i][j]为在以i为根节点的子树中，连接j个用户的最大收益。

u为父节点，v为子节点，那么dp[u][j] = max(dp[u][j],dp[u][j-k]+dp[v][k]+cost)。

实现过程注意加上break，不然容易T。。。

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;
const int maxn = 3005;
const int inf = 0x3f3f3f3f;

struct Nod{
    int b,val,next;
    void init(int b,int val,int next){
        this->b = b;
        this->val = val;
        this->next = next;
    }
}buf[maxn<<1];
int len,E[maxn];
int n,m,dp[maxn][maxn];

void init(){
    int i,j;
    memset(E,-1,sizeof(E));
    len = 0;
    for(i=1;i<=n;i++)for(j=1;j<=n;j++)dp[i][j]=-inf;
}

void add_edge(int a,int b,int val){
    buf[len].init(b,val,E[a]);E[a]=len++;
    buf[len].init(a,val,E[b]);E[b]=len++;
}

void dfs(int u,int pre){
	int i,j,k,v,val;
	for(i=E[u];i!=-1;i=buf[i].next){
		v=buf[i].b;val=buf[i].val;
		if(v == pre) continue;
		dfs(v,u);
		for(j=m;j>=0;j--){
			if(dp[u][j] < -100000) continue;
			for(k=1;k<=m;k++){
				if(dp[v][k] < -100000) break;
				dp[u][j+k]=max(dp[u][j+k],dp[u][j]+dp[v][k]+val);
			}
		}
	}
}

int main(){
    scanf("%d%d",&n,&m);
    init();
    int i,j,k,a,c;
    for(i=1;i<=n-m;i++){
    	scanf("%d",&k);
    	while(k--){
    		scanf("%d%d",&a,&c);
    		add_edge(i,a,-c);
    	}
    }
    for(;i<=n;i++) scanf("%d",&dp[i][1]);
	dfs(1,-1);
	for(i=m;i>=0;i--) if(dp[1][i]>=0) break;
	printf("%d\n",i);
    return 0;
}