[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=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,cnt,tot,ans;
int p[55],L[55],M[55];
int f[55][105][2005];
int g[55][2005],h[55][2005];
char ch[5];
int head[55],deg[55];
struct data{int to,next,v;}e[20005];
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]<0) {
        L[now]=min(L[now],m/M[now]);
        for(int i=0;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]=214748364;
    for(int i=head[now];i>=0;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,-47,sizeof(g));
    g[0][0]=0;
    for(int b=L[now];b>=0;b--) {
        int tot=0;
        for(int i=head[now];i>=0;i=e[i].next) {
            tot++;
            for(int j=0;j<=m;j++)
                for(int k=0;k<=j;k++){
                    g[tot][j]=max(g[tot][j],g[tot-1][j-k]+f[e[i].to][b*e[i].v][k]);}
        }
        for(int i=0;i<=b;i++)
            for(int j=0;j<=m;j++)
                f[now][i][j]=max(f[now][i][j],g[tot][j]+p[now]*(b-i));
    }
}
int main() {
    memset(head,-1,sizeof(head));
    memset(f,-47,sizeof(f));
    n=read(),m=read();
    for(int i=1;i<=n;i++) {
        p[i]=read();
        scanf("%s",ch);
        if(ch[0]=='A') {
            int x=read();
            while(x--) {
                int v=read(),num=read();
                add(i,v,num);
            }
        }
        else M[i]=read(),L[i]=read();
    }
    int sum=0;
    for(int x=1;x<=n;x++) {
        if(!deg[x]) {
            dfs(x);
            sum++;
            for(int i=0;i<=m;i++)
                for(int j=0;j<=i;j++)
                    for(int k=0;k<=L[x];k++)
                        h[sum][i]=max(h[sum][i],h[sum-1][j]+f[x][k][i-j]);
        }
    }
    int ans=0;
    for(int i=0;i<=m;i++) ans=max(ans,h[sum][i]);
    cout<<ans;
}