bzoj 1017: [JSOI2008]魔兽地图DotR【树形dp+背包】

时间:2022-01-01 00:55:27

bzoj上是一个森林啊……?
dp还是太弱了
设f[i][j][k]为到点i,合成j个i并且花费k金币能获得的最大力量值,a[i]为数量上限,b[i]为价格,p[i]为装备力量值
其实这个状态设计出来就好做了,树上背包随便转移一下就行了
因为是森林,所以最后统计答案的时候也要再做一遍背包

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=55;
int n,m,h[N],cnt,ans,p[N],a[N],b[N],f[N][105][2005],g[N][2005],d[N],t[N][2005],tot;
char s[5];
struct qwe
{
    int ne,to,va;
}e[20005];
int read()
{
    int r=0,f=1;
    char p=getchar();
    while(p>'9'||p<'0')
    {
        if(p=='-')
            f=-1;
        p=getchar();
    }
    while(p>='0'&&p<='9')
    {
        r=r*10+p-48;
        p=getchar();
    }
    return r*f;
}
void add(int u,int v,int w)
{
    cnt++;
    e[cnt].to=v;
    e[cnt].ne=h[u];
    e[cnt].va=w;
    d[v]++;
    h[u]=cnt;
}
void dp(int u)
{
    if(!h[u])
    {
        a[u]=min(a[u],m/b[u]);
        for(int i=0;i<=a[u];i++)
            for(int j=i;j<=a[u];j++)
                f[u][i][j*b[u]]=(j-i)*p[u];
        return;
    }
    a[u]=1e9;
    for(int i=h[u];i;i=e[i].ne)
    {
        dp(e[i].to);
        a[u]=min(a[u],a[e[i].to]/e[i].va);
        b[u]+=e[i].va*b[e[i].to];
    }
    a[u]=min(a[u],m/b[u]);
    memset(g,-0x3f,sizeof(g));
    g[0][0]=0;
    for(int l=a[u];l>=0;l--)
    {
        int tot=0;
        for(int i=h[u];i;i=e[i].ne)
        {
            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][l*e[i].va][k]);
        }
        for(int j=0;j<=l;j++)
            for(int k=0;k<=m;k++)
                f[u][j][k]=max(f[u][j][k],g[tot][k]+p[u]*(l-j));
    }
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++)
    {
        p[i]=read();
        scanf("%s",s);
        if(s[0]=='A')
        {
            int x=read();
            while(x--)
            {
                int y=read(),z=read();
                add(i,y,z);
            }
        }
        else 
            b[i]=read(),a[i]=read();
    }
    memset(f,-0x3f,sizeof(f));
    for(int x=1;x<=n;x++)
        if(!d[x])
        {
            dp(x);
            tot++;
            for(int i=0;i<=m;i++)
                for(int j=0;j<=i;j++)
                    for(int k=0;k<=a[x];k++)
                        t[tot][i]=max(t[tot][i],t[tot-1][j]+f[x][k][i-j]);
        }
    for(int i=0;i<=m;i++)
        ans=max(ans,t[tot][i]);
    printf("%d\n",ans);
    return 0;
}