【bzoj2427】[HAOI2010]软件安装

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