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

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;
}