题目大意:有一个大为m的背包,有n个物品,物品有权值vi和质量wi,每一个物品会依赖一个物品,即必须有其依赖的物品才能装入,求最大权值。
m<500,n<100;
题解:一道树上背包模板;
通过依赖建图,然后从儿子向父亲转移即可(会有这么简单吗?)
题目中没有保证没有环,由于一个点只有一个依赖,那么有环的话一定是一个没有出度入读的强连通分量,我们可以用tarjan计算然后dp;
所以建立一个虚点0,将所有入度为0的点(环)连接起来,从0点开始dp即可。
对于实现及树上背包做法:
1、tarjan缩点的实现:最方便的算法是重新建一张图,可以吧原图覆盖掉节省空间;
2、一定要建立虚点,因为树上背包只能从儿子向父亲转移,最方便的求几个不联通的树的方法就是建立虚点。
3、树上背包dp方程:
设dp x,i为在第x个点,容量为i时的最大权值;
首先将正在讨论的点初始化dp数组,设to为所连接点,则:
dp x,i=max(dp x,i ,dp x,i-k+dp to,k) 枚举i,k即可;
可以证明复杂度为O(n*m^2)
代码如下:
#include<bits/stdc++.h>
using namespace std;
const int maxn=(100+10)*5;
const int INF=0x3f3f3f3f;
stack<int>s;
struct Edge{
int x,to,nxt,val;
}e[maxn];
int n,m,w[maxn],v[maxn],d[maxn],head[maxn];
int dfn[maxn],low[maxn],cnt,num;
int w_[maxn],v_[maxn],link[maxn];
int dp[maxn][maxn],in[maxn];
bool vis[maxn];
inline void add(int x,int y)
{
cnt++;e[cnt].to=y;
e[cnt].nxt=head[x],head[x]=cnt;
}
void tarjan(int now)
{
cnt++;low[now]=dfn[now]=cnt;
s.push(now),vis[now]=true;
for(int i=head[now];i;i=e[i].nxt){
int to=e[i].to;
if(!dfn[to]){
tarjan(to);
low[now]=min(low[now],low[to]);
}
else if(vis[i]){
low[now]=min(low[now],dfn[to]);
}
}
if(low[now]==dfn[now]){
num++;int bef;
do{
bef=s.top(),s.pop();
vis[bef]=false;
link[bef]=num;
}while(bef!=now);
}
}
void dfs(int now)
{
for(int i=w_[now];i<=m;i++) dp[now][i]=v_[now];
for(int i=head[now];i;i=e[i].nxt){
int to=e[i].to;
dfs(to);
for(int j=m-w_[now];j>=0;j--)
for(int k=0;k<=j;k++)
dp[now][j+w_[now]]=
max(dp[now][j+w_[now]-k]+dp[to][k],dp[now][j+w_[now]]);;
}
}
int main()
{
freopen("lg2515.in","r",stdin);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&w[i]);
for(int i=1;i<=n;i++) scanf("%d",&v[i]);
for(int i=1;i<=n;i++){
scanf("%d",&d[i]);add(d[i],i);
}
for(int i=1;i<=n;i++)if(!dfn[i])
tarjan(i);
memset(head,0,sizeof(head));cnt=0;
for(int i=1;i<=n;i++)
if(link[i]!=link[d[i]])
add(link[d[i]],link[i]),in[link[i]]++;
for(int i=1;i<=num;i++)
if(!in[i])
add(0,i);
for(int i=1;i<=n;i++)
w_[link[i]]+=w[i],v_[link[i]]+=v[i];
dfs(0);
printf("%d",dp[0][m]);
return 0;
}