hdu 4044 2011北京赛区网络赛E 树形dp ****
专题训练
1 #include<stdio.h> 2 #include<iostream> 3 #include<string.h> 4 #include<algorithm> 5 using namespace std; 6 const int MAXN=1010; 7 const int INF=0x3fffffff; 8 struct Node 9 { 10 int to; 11 int next; 12 }edge[MAXN*2]; 13 int tol; 14 int head[MAXN]; 15 int dp[MAXN][220]; 16 int price[MAXN][60]; 17 int power[MAXN][60]; 18 19 void init() 20 { 21 memset(head,-1,sizeof(head)); 22 tol=0; 23 } 24 void add(int a,int b) 25 { 26 edge[tol].to=b; 27 edge[tol].next=head[a]; 28 head[a]=tol++; 29 edge[tol].to=a; 30 edge[tol].next=head[b]; 31 head[b]=tol++; 32 } 33 int n,m; 34 int tmp[MAXN]; 35 void dfs(int u,int pre) 36 { 37 if(head[u]==-1||(edge[head[u]].to==pre&&edge[head[u]].next==-1))//叶子结点 38 {//叶子结点的条件不要错了!!!! 39 for(int i=0;i<=m;i++)dp[u][i]=0; 40 41 for(int i=0;i<=m;i++)tmp[i]=dp[u][i]; 42 //因为存在价格为0的点。所以倒序的DP并不能保证只取一件物品 43 for(int i=m;i>=0;i--) 44 { 45 for(int j=1;j<=price[u][0];j++) 46 if(price[u][j]<=i) 47 dp[u][i]=max(dp[u][i],tmp[i-price[u][j]]+power[u][j]); 48 49 tmp[i]=dp[u][i];//tmp数组是记录的dp的上一个状态 50 } 51 return; 52 } 53 for(int i=0;i<=m;i++) dp[u][i]=INF; 54 for(int i=head[u];i!=-1;i=edge[i].next) 55 { 56 int v=edge[i].to; 57 if(v==pre)continue; 58 dfs(v,u); 59 for(int j=m;j>=0;j--) 60 { 61 int t=0; 62 for(int k=0;k<=j;k++)//这里k一定要从0开始。 63 t=max(t,min(dp[u][j-k],dp[v][k])); 64 dp[u][j]=t; 65 } 66 } 67 68 69 for(int i=0;i<=m;i++)tmp[i]=dp[u][i]; 70 for(int i=m;i>=0;i--) 71 { 72 for(int j=1;j<=price[u][0];j++) 73 if(price[u][j]<=i) 74 dp[u][i]=max(dp[u][i],tmp[i-price[u][j]]+power[u][j]); 75 //和上面一样分组背包加了个tmp数组 76 tmp[i]=dp[u][i]; 77 } 78 } 79 int main() 80 { 81 int u,v; 82 int T; 83 scanf("%d",&T); 84 while(T--) 85 { 86 init(); 87 scanf("%d",&n); 88 for(int i=1;i<n;i++) 89 { 90 scanf("%d%d",&u,&v); 91 add(u,v); 92 } 93 scanf("%d",&m); 94 for(int i=1;i<=n;i++) 95 { 96 scanf("%d",&price[i][0]); 97 power[i][0]=price[i][0]; 98 for(int j=1;j<=price[i][0];j++) 99 { 100 scanf("%d%d",&price[i][j],&power[i][j]); 101 } 102 } 103 dfs(1,0); 104 printf("%d\n",dp[1][m]); 105 } 106 return 0; 107 }