HAOI 2010 软件安装（Tarjan+树形dp）

【HAOI2010 Day1】软件安装

问题描述

现在我们的手头有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
第3行：V1,V2, … Vi, … ,Vn
第4行：D1,D2, … Di, … ,Dn

输出格式

一个整数，代表最大价值。

样例输入

3 10
5 5 6
2 3 4
0 1 1

样例输出

5

---

此题看上去是裸的树形dp，然而，软件的依赖关系形成的是一个图，这个图可能有多个联通块，还可能存在环，因此需要用Tarjan缩点，然后添加一个虚拟的原点，向每一个缩点后的树根连边，构成一棵树，然后跑dp就好。

关于dp，将每个点看成一个M−Wi的背包，然后再把自己加进去就行了。

---

代码：

#include<stdio.h>#include<iostream>#include<algorithm>#include<cstring>#include<stack>#define N 1005using namespace std;int n,m,W[N],V[N],NW[N],NV[N],D[N];int scc,VT,BE[N],low[N],dfn[N];int TOT,LA[N],NE[N],ST[N],EN[N];int tot,la[N],ne[N],en[N];int F[N][N];bool mark[N],map[N][N];stack<int>S;void ADD(int x,int y){    TOT++;    ST[TOT]=x;    EN[TOT]=y;    NE[TOT]=LA[x];    LA[x]=TOT;}void add(int x,int y){    tot++;    en[tot]=y;    ne[tot]=la[x];    la[x]=tot;}void Tarjan(int u){    int i,v;    dfn[u]=low[u]=++VT;    S.push(u);mark[u]=1;    for(i=LA[u];i;i=NE[i])    {        v=EN[i];        if(!dfn[v])        {            Tarjan(v);            low[u]=min(low[u],low[v]);        }        else if(mark[v])low[u]=min(low[u],dfn[v]);    }    if(low[u]==dfn[u])    {        scc++;        do{            v=S.top();            S.pop();            mark[v]=0;            BE[v]=scc;            NW[scc]+=W[v];            NV[scc]+=V[v];        }while(u!=v);    }}void DP(int x){    int i,j,k,y;    for(i=la[x];i;i=ne[i])    {        y=en[i];        DP(y);        for(j=m-NW[x];j>=0;j--)        for(k=j;k>=0;k--)F[x][j]=max(F[x][j],F[x][k]+F[y][j-k]);    }    for(i=m;i>=0;i--)    if(i>=NW[x])F[x][i]=F[x][i-NW[x]]+NV[x];    else F[x][i]=0;}int main(){    int i,j,x,y,ans=0;    scanf("%d%d",&n,&m);    for(i=1;i<=n;i++)scanf("%d",&W[i]);    for(i=1;i<=n;i++)scanf("%d",&V[i]);    for(i=1;i<=n;i++)    {        scanf("%d",&x);        if(x)ADD(x,i);    }    for(i=1;i<=n;i++)if(!dfn[i])Tarjan(i);    for(i=1;i<=TOT;i++)    {        x=ST[i];y=EN[i];        if(BE[x]!=BE[y])add(BE[x],BE[y]),D[BE[y]]++;    }    BE[0]=scc+1;    for(i=1;i<=scc;i++)if(!D[i])add(BE[0],i);    DP(BE[0]);    for(i=0;i<=m;i++)ans=max(ans,F[BE[0]][i]);    printf("%d",ans);}