洛谷 P2515 [HAOI2010]软件安装

题目描述

现在我们的手头有N个软件，对于一个软件i，它要占用Wi的磁盘空间，它的价值为Vi。我们希望从中选择一些软件安装到一台磁盘容量为M计算机上，使得这些软件的价值尽可能大（即Vi的和最大）。

但是现在有个问题：软件之间存在依赖关系，即软件i只有在安装了软件j（包括软件j的直接或间接依赖）的情况下才能正确工作（软件i依赖软件j)。幸运的是，一个软件最多依赖另外一个软件。如果一个软件不能正常工作，那么它能够发挥的作用为0。

我们现在知道了软件之间的依赖关系：软件i依赖软件Di。现在请你设计出一种方案，安装价值尽量大的软件。一个软件只能被安装一次，如果一个软件没有依赖则Di=0，这时只要这个软件安装了，它就能正常工作。

输入输出格式

输入格式：
第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 ）

输出格式：
一个整数，代表最大价值

输入输出样例

输入样例#1：
3 10
5 5 6
2 3 4
0 1 1
输出样例#1：
5

---

【分析】
强连通分量缩点+树形分组背包dp

---

【代码】

//洛谷 [HAOI2010]软件安装
#include<stack>
#include<queue> 
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=105;
stack <int> s;
struct edge {int to,next;} f[10*mxn];
int head[mxn],dfn[mxn],low[mxn];
int wei[mxn],val[mxn],be[mxn],dp[mxn][505];  //节点u，选了重量为j，得到的最大价值 
struct wu {int w,v,d;} a[mxn];
bool in[mxn],con[mxn][mxn],du[mxn];
int n,m;
int cnt,tim,num;
inline void add(int u,int v)
{
    f[++cnt].to=v,f[cnt].next=head[u],head[u]=cnt;
}
inline void tarjan(int u)
{
    int i,v;
    dfn[u]=low[u]=++tim;
    in[u]=1;
    s.push(u);
    for(i=head[u];i;i=f[i].next)
    {
        v=f[i].to;
        if(!dfn[v]) tarjan(v),low[u]=min(low[u],low[v]);
        else if(in[v]) low[u]=min(low[u],dfn[v]);
    }
    if(dfn[u]==low[u])
    {
        num++;
        do
        {
            v=s.top();
            s.pop();
            in[v]=0;
            be[v]=num;
            wei[num]+=a[v].w;
            val[num]+=a[v].v;
        }while(u!=v);
    }
}
inline void dfs(int u)
{
    for(int k=wei[u];k<=m;k++) dp[u][k]=val[u];
    for(int i=head[u];i;i=f[i].next)
    {
        int v=f[i].to;
        dfs(v);
        for(int j=m-wei[u];j>=0;j--)
          for(int k=0;k<=j;k++)
            dp[u][j+wei[u]]=max(dp[u][j+wei[u]],dp[u][j+wei[u]-k]+dp[v][k]);
    }
}
int main()
{
    int i,j,k,u,v,d,ans=0;
    scanf("%d%d",&n,&m);
    fo(i,1,n) scanf("%d",&a[i].w);
    fo(i,1,n) scanf("%d",&a[i].v);
    fo(i,1,n) scanf("%d",&a[i].d),add(a[i].d,i);
    fo(i,0,n) if(!dfn[i]) tarjan(i);
    memset(head,0,sizeof head);cnt=0;
    fo(i,1,n)
    {
        u=a[i].d,v=i;
        if(be[u]!=be[v]) add(be[u],be[v]),du[be[v]]=1;
    }
    fo(i,1,num) if(!du[i]) add(0,i);
    dfs(0);
    printf("%d\n",dp[0][m]);
    return 0;
}