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;
}