树上dp——软件安装

最新推荐文章于 2024-09-14 20:30:05 发布

green qwq

最新推荐文章于 2024-09-14 20:30:05 发布

阅读量822

点赞数 11

分类专栏：洛谷算法题合集文章标签：算法数据结构

本文链接：https://blog.csdn.net/m0_60738889/article/details/137387231

版权

洛谷算法题合集专栏收录该内容

142 篇文章 0 订阅

订阅专栏

[HAOI2010] 软件安装

题目描述

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

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

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

输入格式

第 1 行： $N,M(0\leq N\leq 100, 0\leq M\leq 500)$

第 2 行： $W_1,W_2, ... W_i, ..., W_n (0\leq W_i\leq M)$

第 3 行： $V_1, V_2, ..., V_i, ..., V_n (0\leq V_i\leq 1000)$

第 4 行： $D_1, D_2, ..., D_i, ..., D_n (0\leq D_i\leq N, D_i≠i)$

输出格式

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

样例 #1

样例输入 #1

样例输出 #1

思路

首先，对于本道题，题目描述中并没有说这个东西一定是一棵树,它有可能是一张带环图，也可能是一个森林，因此我们得缩点。
我们发现:这个图建完后是一个带环图。1，2，3号节点互为依赖条件。一旦有其中任何一个点没有被选择，整个环里面的所有内容都无法被使用。我们重新观察这张图，我们发现1，2，3号节点的共性：我们可以对它整个选择，也可以对它整个不选择，对其中的任何一个真子集都没有意义。对于环中点有共性的图，我们考虑用缩点来解决问题。
也就是说我如果要选联通分量内的一个，那么整个联通分量都得选（核心）

代码

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N = 220,M = 1010;

int e[N],h[N],hs[N],w[N],ne[N],idx;
int dfn[N],low[N];
int v[N];
bool in_stk[N];
int stk[N],top,scc_size[N];
int scc_cnt,id[N],timecuocuo;
int din[N];
int scc_v[N],scc_w[N];
int f[N][M];//以i为根，体积为j，值为f[i][j]
int n,m;

void add(int h[],int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void targin(int u){
    low[u]=dfn[u]=++timecuocuo;
    stk[++top]=u,in_stk[u]=true;
    
    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        
        if(!dfn[j]){
            targin(j);
            low[u]=min(low[u],low[j]);
        }else if(in_stk[j]){
            low[u]=min(low[u],dfn[j]);
        }
    }
    
    if(low[u]==dfn[u]){
        scc_cnt++;
        int y;
        do{
            y=stk[top--];
            in_stk[y]=false;
            id[y]=scc_cnt;
            scc_size[scc_cnt]++;
        }while(y!=u);
    }
}

void dfs(int u){
    for(int i=m;i>=scc_w[u];i--){
        f[u][i]=scc_v[u];
    }
    
    for(int i=hs[u];~i;i=ne[i]){
        int p=e[i];
        
        dfs(p);
        
        for(int j=m;j>=scc_w[u];j--){
            for(int k=0;k<=j-scc_w[u];k++){//选根节点必须滴
                f[u][j]=max(f[u][j],f[p][k]+f[u][j-k]);
            }
        }
    }
}

int main(){
    cin>>n>>m;
    memset(h,-1,sizeof h);
    memset(hs,-1,sizeof hs);
    for(int i=1;i<=n;i++)cin>>w[i];
    
    for(int i=1;i<=n;i++)cin>>v[i];
    
    for(int i=1;i<=n;i++){
        int b;
        cin>>b;
        if(b!=0)
        add(h,b,i);
    }
    
    for(int i=1;i<=n;i++){
        if(!dfn[i]){
            targin(i);
        }
    }
    
    for(int i=1;i<=n;i++){
        for(int j=h[i];~j;j=ne[j]){
            int k=e[j];
            int a=id[i],b=id[k];
            if(a!=b){
                add(hs,a,b);
                din[b]++;
            }
        }
        scc_v[id[i]]+=v[i];
        scc_w[id[i]]+=w[i];
    }
    //超级原点
    for(int i=1;i<=scc_cnt;i++){
        if(!din[i]){
            add(hs,0,i);
        }
    }
    
    dfs(0);
    
    cout<<f[0][m];
    
    return 0;
}