hdu 4044 2011北京赛区网络赛E 树形dp ****

专题训练

 #include<stdio.h>
 #include<iostream>
 #include<string.h>
 #include<algorithm>
 using namespace std;
 const int MAXN=;
 const int INF=0x3fffffff;
 struct Node
 {
     int to;
     int next;
 }edge[MAXN*];
 int tol;
 int head[MAXN];
 int dp[MAXN][];
 int price[MAXN][];
 int power[MAXN][];

 void init()
 {
     memset(head,-,sizeof(head));
     tol=;
 }
 void add(int a,int b)
 {
     edge[tol].to=b;
     edge[tol].next=head[a];
     head[a]=tol++;
     edge[tol].to=a;
     edge[tol].next=head[b];
     head[b]=tol++;
 }
 int n,m;
 int tmp[MAXN];
 void dfs(int u,int pre)
 {
     if(head[u]==-||(edge[head[u]].to==pre&&edge[head[u]].next==-))//叶子结点
     {//叶子结点的条件不要错了！！！！
         for(int i=;i<=m;i++)dp[u][i]=;

         for(int i=;i<=m;i++)tmp[i]=dp[u][i];
 //因为存在价格为0的点。所以倒序的DP并不能保证只取一件物品
         for(int i=m;i>=;i--)
         {
             for(int j=;j<=price[u][];j++)
                if(price[u][j]<=i)
                  dp[u][i]=max(dp[u][i],tmp[i-price[u][j]]+power[u][j]);

             tmp[i]=dp[u][i];//tmp数组是记录的dp的上一个状态
         }
         return;
     }
     for(int i=;i<=m;i++) dp[u][i]=INF;
     for(int i=head[u];i!=-;i=edge[i].next)
     {
         int v=edge[i].to;
         if(v==pre)continue;
         dfs(v,u);
         for(int j=m;j>=;j--)
         {
             int t=;
             for(int k=;k<=j;k++)//这里k一定要从0开始。
               t=max(t,min(dp[u][j-k],dp[v][k]));
             dp[u][j]=t;
         }
     }

     for(int i=;i<=m;i++)tmp[i]=dp[u][i];
     for(int i=m;i>=;i--)
     {
         for(int j=;j<=price[u][];j++)
           if(price[u][j]<=i)
              dp[u][i]=max(dp[u][i],tmp[i-price[u][j]]+power[u][j]);
         //和上面一样分组背包加了个tmp数组
         tmp[i]=dp[u][i];
     }
 }
 int main()
 {
     int u,v;
     int T;
     scanf("%d",&T);
     while(T--)
     {
         init();
         scanf("%d",&n);
         for(int i=;i<n;i++)
         {
             scanf("%d%d",&u,&v);
             add(u,v);
         }
         scanf("%d",&m);
         for(int i=;i<=n;i++)
         {
             scanf("%d",&price[i][]);
             power[i][]=price[i][];
             for(int j=;j<=price[i][];j++)
             {
                 scanf("%d%d",&price[i][j],&power[i][j]);
             }
         }
         dfs(,);
         printf("%d\n",dp[][m]);
     }
     return ;
 }