HYSBZ - 1017 魔兽地图【树形DP依赖性背包】

HYSBZ - 1017

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

题意：见上文
分析：树形DP。 dp[i][j][k]表示以第i种物品为根的子树花费j，向上层提供k个i所得到的的最大贡献值。
    对于每个节点，枚举合成1至n个物品的最大能力值，背包问题。
ACcode

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int INF=0x3f3f3f3f;
const int N=2020;
const int M=58;
int dp[M][105][N],pre[M],num[M],val[M],f[M][N];
int n,m;
bool fa[M],flag;
struct edge
{
    int to,need;
    edge(int x,int y)
    {
        to=x;
        need=y;
    }
};
vector<edge> g[M];
void dfs(int x)
{
    if(g[x].size()==0)
    {   //cout<<"  *"<<x<<endl;
        num[x]=min(num[x],m/pre[x]);
        for(int i=0;i<=num[x];i++)
            for(int j=i;j<=num[x];j++)
            dp[x][i][j*pre[x]]=(j-i)*val[x];
            return ;
    }
    num[x]=INF;
    for(int i=0;i<g[x].size();i++)
    {
        int y=g[x][i].to,z=g[x][i].need;
        dfs(y);
        num[x]=min(num[x],num[y]/z);
        pre[x]+=(z*pre[y]);
    }
    num[x]=min(num[x],m/pre[x]);
    memset(f,-0x3f3f3f3f,sizeof(f));
    f[0][0]=0;  //背包
    for(int c=num[x];c>=0;c--)
    {
        for(int i=0;i<g[x].size();i++)
        {
            int y=g[x][i].to,z=g[x][i].need;
            for(int v=0;v<=m;v++)
                for(int k=v;k>=0;k--)
                f[i+1][v]=max(f[i+1][v],f[i][v-k]+dp[y][c*z][k]);
        }
        int len=g[x].size();
        for(int i=0;i<=c;i++)
            for(int v=0;v<=m;v++)
            dp[x][i][v]=max(dp[x][i][v],(c-i)*val[x]+f[len][v]);
    }
    return ;
}
void init()
{
    memset(dp,-0x3f3f3f3f,sizeof(dp));
    memset(fa,false,sizeof(fa));
    memset(pre,0,sizeof(pre));
    for(int i=1;i<=n;i++)
        g[i].clear();
}
char ch[2];
int main()
{
    int ans;
   while(scanf("%d%d",&n,&m)!=EOF)
   {
       init();
       ans=-INF;
        for(int i=1;i<=n;i++)
        {
            scanf("%d%s",&val[i],ch);
            if(ch[0]=='A')
            {
                int tot;
                scanf("%d",&tot);
                while(tot--)
                {
                    int to,vall;
                    scanf("%d%d",&to,&vall);
                    fa[to]=true;
                    g[i].push_back(edge(to,vall));
                }
            }
            else scanf("%d%d",&pre[i],&num[i]);
        }
       flag=false;
       for(int i=1;i<=n;i++)if(fa[i])flag=true;
       if(flag)
       {
           for(int i=1;i<=n;i++)
           {
               if(fa[i])continue;
               dfs(i);
               for(int j=0;j<=num[i];j++)
                for(int k=0;k<=m;k++)
                ans=max(ans,dp[i][j][k]);
           }
       }
       else
       {
           memset(f,0,sizeof(f));
           for(int i=1;i<=n;i++)
           {
               for(int j=0;j<=num[i];j++)
               for(int v=m;v>=j*pre[i];v--)
               {
                f[i][v]=max(f[i][v],f[i-1][v-j*pre[i]]+j*val[i]);
               }
           }
           for(int v=0;v<=m;v++)
            ans=max(ans,f[n][v]);
       }
       printf("%d\n",ans);
   }
   return 0;
}