[HAOI2010]软件安装

Tarjan+树形DP

[HAOI2010]软件安装
根据题意，如果依赖关系构成一个环，那只有环里所有软件都安装才能使其价值不为0；

在入度为0的点之上加一个点，这样可以一下DP完

然后用树状DP解决，注意到如果下面的点的占用空间超过m，其价值在DP数组中直接化为0；其实是一个树上的01背包

---

#include<iostream>
#include<cstdio>
#include<stack>
using namespace std;

stack<int> s;
int n,m,weight[200],value[200];
bool indegree[600];
int LOW[600],DFN[600],steps,total;
bool instack[600];
int dp[200][600];

struct E{
    int next,to,from;
}edge[600],newedge[600];
int head[600],nhead[600],edge_num,newedge_num;
struct S{
    int w,v;
}scc[600];
int belong[600];

void addedge(int x,int y,int z){
    if(z==1){
        edge[++edge_num].next=head[x];
        edge[edge_num].to=y;
        edge[edge_num].from=x;
        head[x]=edge_num;
    }
    if(z==2){
        newedge[++newedge_num].next=nhead[x];
        newedge[newedge_num].to=y;
        newedge[newedge_num].from=x;
        nhead[x]=newedge_num;
    }
}

void Tarjan(int x){
    int i;
    s.push(x);instack[x]=true;
    DFN[x]=LOW[x]=++steps;
    for(i=head[x];i;i=edge[i].next){
        if(!DFN[edge[i].to]){
            Tarjan(edge[i].to);
            LOW[x]=min(LOW[x],LOW[edge[i].to]);
        }
        if(instack[edge[i].to])
            LOW[x]=min(LOW[x],DFN[edge[i].to]);
    }
    if(LOW[x]==DFN[x]){
        int t;
        total++;
        do{
            t=s.top();s.pop();
            instack[t]=false;
            belong[t]=total;
            scc[total].v+=value[t];
            scc[total].w+=weight[t];
        }
        while(t!=x);
    }
}

void Newmap(){
    int i,j;
    for(i=1;i<=n;i++){
        for(j=head[i];j;j=edge[j].next){
            if(belong[i]!=belong[edge[j].to]){
                indegree[belong[edge[j].to]]=true;
                addedge(belong[i],belong[edge[j].to],2);
            }
        }
    }
//  for(i=1;i<=edge_num;i++){
//      if(belong[edge[i].from]!=belong[edge[i].to]){
//          indegree[belong[edge[i].to]]=true;
//          addedge(belong[edge[i].from],belong[edge[i].to],2);
//      }
//  }
}


//此处DP为树上01背包 
void DP(int x){//树上DP ，x为当前(根)节点 
    int i,j,k;
    for(i=nhead[x]; i ;i=newedge[i].next){//节点x连接的每一条边 
        DP(newedge[i].to);//从延伸出去的点继续DFS 
        for(j=m-scc[x].w;j>=0;j--){// scc[x].w为当前节点的空间值(是题目中的“空间”的含义) 
            //m -scc[x].w就是刨除该节点不算，可以供安装的剩余空间 
            for(k=0;k<=j;k++){
                //这里是递推
                //k为选择的空间量，j-k是仍可供选择的空间 
                dp[x][j]=max(dp[x][j],dp[x][k]+dp[newedge[i].to][j-k]);
            }
        }
    }
    //上面搞完了，或者上面没有连到节点的边
    //到底了的情况很自然就是这样
    //要是没到底，如果 scc[x].w 超过 j ，就说明 j 的容量不足以安装下面的子树，以x为根的树价值为0 
    for(j=m;j>=0;j--){
        if(j>=scc[x].w)
            dp[x][j]=dp[x][j-scc[x].w]+scc[x].v;
        else
            dp[x][j]=0;
    }
}

int main(){
    freopen("2515.in","r",stdin);
    scanf("%d%d",&n,&m);
    int i;
    for(i=1;i<=n;i++)
        scanf("%d",&weight[i]);
    for(i=1;i<=n;i++)
        scanf("%d",&value[i]);
    for(i=1;i<=n;i++){
        int k;
        scanf("%d",&k);
        if(k==0) continue;
        addedge(k,i,1);
    }
    for(i=1;i<=n;i++){
        if(!DFN[i])
            Tarjan(i);
    }
    Newmap();
    for(i=1;i<=total;i++){
        if(!indegree[i]){
            indegree[i]=true;
            addedge(total+1,i,2);
        }
    }
    DP(total+1);
    printf("%d",dp[total+1][m]);
    return 0;
}