[HAOI2010]软件安装

简单的tarjan+（本蒟蒻刚刚接触不久）恶心的树形DP

题面

题目描述
现在我们的手头有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

---

解题思路

有可能出现依赖关系的环，我们把环tarjan缩成点，因为你选择这个环中的任意一个软件都需要把这个环整个都安装，所以直接累加这个环的w和v缩点。
暴力重新建图后，整个就形成了一个森林，为了方便，建一个虚的树根连上每个入度为0的点，然后在树上跑DP求解。
DP：设f[i][j]表示i节点容量为j所得的最大价值，由它的儿子转移过来

空间，常数巨大的代码

# include <stdio.h>
# include <stdlib.h>
# include <iostream>
# include <string.h>
using namespace std;

# define N 501
# define IL inline
# define RG register
# define UN unsigned
# define ll long long
# define oo 2147483647
# define mem(a, b) memset(a, b, sizeof(a))
# define max(a, b) ((a) > (b)) ? (a) : (b)
# define min(a, b) ((a) < (b)) ? (a) : (b)

IL int Get(){
    RG char c = '!'; RG int num = 0, z = 1;
    while(c != '-' && (c > '9' || c < '0')) c = getchar();
    if(c == '-') z = -1, c = getchar();
    while(c >= '0' && c <= '9') num = num * 10 + c - '0', c = getchar();
    return num * z;
}

int n, w[N], m, v[N], ft[N], to[N], nt[N], cnt, f[N][501];
int dfn[N], low[N], bg[N], isk[N], sk[N], cnt1, num, r[N];

IL void Add(RG int u, RG int vv){
    nt[cnt] = ft[u]; to[cnt] = vv; ft[u] = cnt++;
}

IL void Dfs(RG int u){
    dfn[u] = low[u] = ++cnt1;
    isk[u] = 1; sk[++sk[0]] = u;
    for(RG int e = ft[u]; e != -1; e = nt[e]){
        RG int vv = to[e];
        if(!dfn[vv]){
            Dfs(vv);
            low[u] = min(low[u], low[vv]);
        }
        else if(isk[vv]) low[u] = min(low[u], dfn[vv]);
    }
    if(dfn[u] == low[u]){
        RG int vv = sk[sk[0]--]; isk[vv] = 0;
        bg[vv] = ++num; w[num] += w[vv]; v[num] += v[vv];
        while(u != vv){
            vv = sk[sk[0]--];
            isk[vv] = 0;
            bg[vv] = num; w[num] += w[vv]; v[num] += v[vv];
        }
    }
}

IL void DP(RG int u){
    for(RG int j = m; j >= w[u]; j--)
        f[u][j] = v[u];
    for(RG int e = ft[u]; e != -1; e = nt[e]){
        RG int vv = to[e];
        DP(to[e]);
        for(RG int j = m - w[u]; j >= 0; j--)
            for(RG int k = 0; k <= j; k++)
                f[u][j + w[u]] = max(f[u][j + w[u]], f[u][j + w[u] - k] + f[vv][k]);
    }
}

int main(){
    mem(ft, -1);
    n = Get(); m = Get();
    for(RG int i = 1; i <= n; i++)
        w[i] = Get();
    for(RG int i = 1; i <= n; i++)
        v[i] = Get();
    for(RG int i = 1; i <= n; i++){
        RG int u = Get();
        if(u) Add(u, i);
    }
    num = n;
    for(RG int i = 1; i <= n; i++)
        if(!dfn[i]) Dfs(i);
    for(RG int i = 1; i <= n; i++)
        for(RG int e = ft[i]; e != -1; e = nt[e]){
            RG int vv = to[e];
            if(bg[i] != bg[vv]) Add(bg[i], bg[vv]), r[bg[vv]]++;
        }
    for(RG int i = n + 1; i <= num; i++)
        if(!r[i]) Add(num + 1, i);
    DP(num + 1);
    printf("%d\n", f[num + 1][m]);
    return 0;
}

我不会告诉你，这道题我写萎了交了n遍才A

转载于:https://www.cnblogs.com/cjoieryl/p/8206423.html