[BZOJ1017][JSOI2008]魔兽地图DotR 树形dp

1017: [JSOI2008]魔兽地图DotR

Time Limit: 30 Sec  Memory Limit: 162 MB
Submit: 2597  Solved: 1010
[Submit][Status][Discuss]

Description

　　DotR (Defense of the Robots) Allstars是一个风靡全球的魔兽地图，他的规则简单与同样流行的地图DotA 
(Defense of the Ancients) Allstars。DotR里面的英雄只有一个属性——力量。他们需要购买装备来提升自己的
力量值，每件装备都可以使佩戴它的英雄的力量值提高固定的点数，所以英雄的力量值等于它购买的所有装备的力
量值之和。装备分为基本装备和高级装备两种。基本装备可以直接从商店里面用金币购买，而高级装备需要用基本
装备或者较低级的高级装备来合成，合成不需要附加的金币。装备的合成路线可以用一棵树来表示。比如，Sange 
and Yasha的合成需要Sange,Yasha和Sange and Yasha Recipe Scroll三样物品。其中Sange又要用Ogre Axe, Belt
 of Giant Strength和 Sange Recipe Scroll合成。每件基本装备都有数量限制，这限制了你不能无限制地合成某
些性价比很高的装备。现在，英雄Spectre有M个金币，他想用这些钱购买装备使自己的力量值尽量高。你能帮帮他
吗？他会教你魔法Haunt（幽灵附体）作为回报的。

Input

　　第一行包含两个整数，N (1 <= n <= 51) 和 m (0 <= m <= 2,000)。分别表示装备的种类数和金币数。装备
用1到N的整数编号。接下来的N行，按照装备1到装备n的顺序，每行描述一种装备。每一行的第一个非负整数表示这
个装备贡献的力量值。接下来的非空字符表示这种装备是基本装备还是高级装备，A表示高级装备，B表示基本装备
。如果是基本装备，紧接着的两个正整数分别表示它的单价（单位为金币）和数量限制（不超过100）。如果是高
级装备，后面紧跟着一个正整数C，表示这个高级装备需要C种低级装备。后面的2C个数，依次描述某个低级装备的
种类和需要的个数。

Output

　　第一行包含一个整数S，表示最多可以提升多少点力量值。

Sample Input

10 59
5 A 3 6 1 9 2 10 1
1 B 5 3
1 B 4 3
1 B 2 3
8 A 3 2 1 3 1 7 1
1 B 5 3
5 B 3 3
15 A 3 1 1 5 1 4 1
1 B 3 5
1 B 4 3

Sample Output

33

对于这道题，我们设f[i][j][k]表示处理到第i个点，向父亲节点合并j个装备，花了k元钱的最大力量值（不包括合并的j个）。

设g[i][j]表示当前节点的前i个子树花了j元钱的最大力量（不包括合并的）。

我们先枚举购买当前装备的总数量b，

对于f的转移，我们枚举j，k，有转移方程f[i][j][k]=max{g[i][k]+p[i]*(b-j)}

对于g的转移，我们枚举子树tot，钱数j和当前子树的钱数k，有g[tot][j]=max{g[tot-1][j-k]+f[i][b*e[i].v][k]}

之后统计答案即可

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<cstdlib>
 #include<cmath>
 #include<algorithm>
 using namespace std;
 int read() {
     int x=,f=;char ch=getchar();
     while(!isdigit(ch)){if(ch=='-') f=-;ch=getchar();}
     while(isdigit(ch)){x=x*+ch-'';ch=getchar();}
     return x*f;
 }
 int n,m,cnt,tot,ans;
 int p[],L[],M[];
 int f[][][];
 int g[][],h[][];
 char ch[];
 int head[],deg[];
 struct data{int to,next,v;}e[];
 void add(int u,int v,int w){e[cnt].to=v;e[cnt].next=head[u];head[u]=cnt;e[cnt].v=w;cnt++;deg[v]++;}
 void dfs(int now) {
     if(head[now]<) {
         L[now]=min(L[now],m/M[now]);
         for(int i=;i<=L[now];i++)
             for(int j=i;j<=L[now];j++) f[now][i][j*M[now]]=(j-i)*p[now];
         return ;
     }
     L[now]=;
     for(int i=head[now];i>=;i=e[i].next) {
         int to=e[i].to;
         dfs(to);
         L[now]=min(L[now],L[to]/e[i].v);
         M[now]+=e[i].v*M[to];
     }
     L[now]=min(L[now],m/M[now]);
     memset(g,-,sizeof(g));
     g[][]=;
     for(int b=L[now];b>=;b--) {
         int tot=;
         for(int i=head[now];i>=;i=e[i].next) {
             tot++;
             for(int j=;j<=m;j++)
                 for(int k=;k<=j;k++){
                     g[tot][j]=max(g[tot][j],g[tot-][j-k]+f[e[i].to][b*e[i].v][k]);}
         }
         for(int i=;i<=b;i++)
             for(int j=;j<=m;j++)
                 f[now][i][j]=max(f[now][i][j],g[tot][j]+p[now]*(b-i));
     }
 }
 int main() {
     memset(head,-,sizeof(head));
     memset(f,-,sizeof(f));
     n=read(),m=read();
     for(int i=;i<=n;i++) {
         p[i]=read();
         scanf("%s",ch);
         if(ch[]=='A') {
             int x=read();
             while(x--) {
                 int v=read(),num=read();
                 add(i,v,num);
             }
         }
         else M[i]=read(),L[i]=read();
     }
     int sum=;
     for(int x=;x<=n;x++) {
         if(!deg[x]) {
             dfs(x);
             sum++;
             for(int i=;i<=m;i++)
                 for(int j=;j<=i;j++)
                     for(int k=;k<=L[x];k++)
                         h[sum][i]=max(h[sum][i],h[sum-][j]+f[x][k][i-j]);
         }
     }
     int ans=;
     for(int i=;i<=m;i++) ans=max(ans,h[sum][i]);
     cout<<ans;
 }