Description
现在我们的手头有N个软件,对于一个软件i,它要占用Wi的磁盘空间,它的价值为Vi。我们希望从中选择一些软件安装到一台磁盘容量为M计算机上,使得这些软件的价值尽可能大(即Vi的和最大)。
但是现在有个问题:软件之间存在依赖关系,即软件i只有在安装了软件j(包括软件j的直接或间接依赖)的情况下才能正确工作(软件i依赖软件j)。幸运的是,一个软件最多依赖另外一个软件。如果一个软件不能正常工作,那么它能够发挥的作用为0。
我们现在知道了软件之间的依赖关系:软件i依赖软件Di。现在请你设计出一种方案,安装价值尽量大的软件。一个软件只能被安装一次,如果一个软件没有依赖则Di=0,这时只要这个软件安装了,它就能正常工作。
Input
第1行:N, M (0<=N<=100, 0<=M<=500)
第2行:W1, W2, … Wi, …, Wn (0<=Wi<=M )
第3行:V1, V2, …, Vi, …, Vn (0<=Vi<=1000 )
第4行:D1, D2, …, Di, …, Dn (0<=Di<=N, Di≠i )
Output
一个整数,代表最大价值。
Sample Input
3 10
5 5 6
2 3 4
0 1 1
Sample Output
5
题解
tarjan缩点,树上DP,DP时注意从大到小枚举。f(i,j)表示对i及其子树花费j代价产生的最大价值
代码
#include<bits/stdc++.h>
#define inf 0x7fffffff
#define N 20005
#define M 400005
#define ll long long
using namespace std;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int n,m,tot,scc,ind,top;
int v[105],w[105];
int sv[105],sw[105];
int dfn[105],low[105],bl[105];
int stk[105],f[105][505],in[505];
bool flag[105];
int Head[105],Head2[105],ret[505],ret2[505];
int Next[505],Next2[505];
inline void ins1(int u,int v)
{
ret[++tot]=v;Next[tot]=Head[u];Head[u]=tot;
}
inline void ins2(int u,int v)
{
in[v]=1;ret2[++tot]=v;Next2[tot]=Head2[u];Head2[u]=tot;
}
void tarjan(int u)
{
low[u]=dfn[u]=++ind;
flag[u]=1;stk[++top]=u;
for (int i=Head[u];i;i=Next[i])
if (dfn[ret[i]]==0)
{
tarjan(ret[i]);
low[u]=min(low[u],low[ret[i]]);
}
else if (flag[ret[i]]) low[u]=min(low[u],low[ret[i]]);
if (low[u]==dfn[u])
{
scc++;int now=0;
while (now!=u)
{
now=stk[top--];flag[now]=0;
bl[now]=scc;
sw[scc]+=w[now];
sv[scc]+=v[now];
}
}
}
void build()
{
tot=0;
for (int x=1;x<=n;x++)
for (int i=Head[x];i;i=Next[i])
if (bl[ret[i]]!=bl[x])
ins2(bl[x],bl[ret[i]]);
}
void dp(int u)
{
if (flag[u]) return;flag[u]=1;
for (int i=Head2[u];i;i=Next2[i])
{
dp(ret2[i]);
for (int j=m-sw[u];j>=0;j--)
for (int k=j;k;k--)
f[u][j]=max(f[u][j],f[ret2[i]][k]+f[u][j-k]);
}
for (int j=m;j;j--)
if (j>=sw[u]) f[u][j]=f[u][j-sw[u]]+sv[u];
else f[u][j]=0;
}
int main()
{
n=read();m=read();
for (int i=1;i<=n;i++) w[i]=read();
for (int i=1;i<=n;i++) v[i]=read();
for (int i=1;i<=n;i++)
{
int x=read();
if (x) ins1(x,i);
}
for (int i=1;i<=n;i++)
if (!dfn[i]) tarjan(i);
build();
for (int i=1;i<=scc;i++)
if (!in[i]) ins2(scc+1,i);
dp(scc+1);
printf("%d",f[scc+1][m]);
return 0;
}