[JSOI2008]魔兽地图DotR
题意
给一棵树,选取叶节点一次需要花费\(w_i\)代价,获得\(v_i\)收益,一个叶子节点最多选择\(l_i\)次,非叶子节点也有收益\(v_i\),它由其叶子节点按照一定比例混合得来。问花费为\(m\)所能获得的最大收益
思路
神仙树形dp
首先看出来是树形dp,之后就一定是树上背包啦~~~
设\(f[i][j][k]\)表示在\(i\)号节点,向父亲传\(j\)次,且\(i\)子树共花费了\(k\)代价的最大收益
先枚举当前节点的选取次数\(l\)
用\(g[i][j]\)表示考虑了当前子树的前\(i\)个儿子,且它们总花费代价为\(j\)的最大收益,转移方程为\(g[i][j]=max(g[i-1][j-k]+f[v][l*e][k])\),得到\(g[tot][j]\)(即所有儿子,其中e为v儿子转换成父亲的比例)
再枚举当前节点向父亲传的次数\(j\)
有\(f[rt][j][k]=max(g[tot][k]+v[rt]*(l-j))\)
另外,由于可能不止一棵树,需要用背包对森林进行合并,这个操作和01背包相同
Code
#include<bits/stdc++.h>
#define N 55
#define M 2005
#define Max(x,y) ((x)>(y) ? (x):(y))
using namespace std;
int n,m;
int S[N],W[N],L[N];//力量,价格,数量
int rd[N];
int f[N][105][M];//在i节点选择j个上传,子树花费k代价的总力量值
int ans[M],g[N][M],tot;
struct Edge
{
int next,to,val;//val是使用次数
}edge[N*N];int head[N],cnt;
void add_edge(int from,int to,int val)
{
edge[++cnt].next=head[from];
edge[cnt].to=to;
edge[cnt].val=val;
head[from]=cnt;
}
template <class T>
void read(T &x)
{
char c;int sign=1;
while((c=getchar())>'9'||c<'0') if(c=='-') sign=-1; x=c-48;
while((c=getchar())>='0'&&c<='9') x=x*10+c-48; x*=sign;
}
void dfs(int rt)//处理rt子树
{
if(!head[rt])//叶子
{
L[rt]=min(L[rt],m/W[rt]);
for(int i=0;i<=L[rt];++i)
for(int j=0;j<=i;++j)
f[rt][j][i*W[rt]]=(i-j)*S[rt];
return;
}
L[rt]=100000000;
for(int i=head[rt];i;i=edge[i].next)
{
int v=edge[i].to;
dfs(v);
L[rt]=min(L[rt],L[v]/edge[i].val);//判边界
W[rt]+=W[v]*edge[i].val;
}
L[rt]=min(L[rt],m/W[rt]);//神装限制数量
memset(g,-50,sizeof(g));
g[0][0]=0;
for(int l=L[rt];l>=0;--l)
{
tot=0;
for(int i=head[rt];i;i=edge[i].next)
{
int v=edge[i].to;
++tot;
for(int j=0;j<=m;++j)//前tot个儿子共花费j
for(int k=0;k<=j;++k)//前tot-1个花费k
{
g[tot][j]=Max(g[tot][j],g[tot-1][k]+f[v][l*edge[i].val][j-k]);
}
}
for(int j=0;j<=l;++j)//上传j个
for(int k=0;k<=m;++k)
f[rt][j][k]=Max(f[rt][j][k],g[tot][k]+S[rt]*(l-j));
}
}
int main()
{
memset(f,-50,sizeof(f));
read(n);read(m);
for(int i=1;i<=n;++i)
{
char op[2];
read(S[i]);
scanf("%s",op);
if(op[0]=='A')//神装
{
int c; read(c);
for(int j=1;j<=c;++j)
{
int x; read(x);
int p; read(p);
add_edge(i,x,p);
++rd[x];
}
}
else read(W[i]),read(L[i]);
}
for(int i=1;i<=n;++i)
{
if(!rd[i])
{
dfs(i);
for(int k=m;k>=0;--k)
for(int j=0;j<=k;++j)
ans[k]=Max(ans[k],ans[k-j]+f[i][0][j]);
}
}
cout<<ans[m]<<endl;
return 0;
}