[BZOJ]2427 软件安装 O(nm)树形背包

2427: [HAOI2010]软件安装

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 1913  Solved: 757
[ Submit][ Status][ Discuss]

Description

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

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

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

Input

第1行：N, M  （0<=N<=100, 0<=M<=500）
      第2行：W₁, W₂, ... W_i, ..., W_n （0<=W_i<=M ）
      第3行：V₁, V₂, ..., V_i, ..., V_n  （0<=Vi<=1000 ）
      第4行：D₁, D₂, ..., D_i, ..., D_n （0<=D_i<=N, D_i≠i ）

Output

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

Sample Input

3 10
5 5 6
2 3 4
0 1 1

Sample Output

5

HINT

Source

Day2

[ Submit][ Status][ Discuss]

HOME Back

　　今天读了徐持衡的《浅谈几类背包题》， 学习了其中树形dp的O(nm)做法. 具体做法大家可以看看论文， 我在这里就不作细述了.

　　这道题裸的树上背包， 但注意可能有环， 而且可能是多个图. 所以要tarjan缩点重新建图. 然后让一个虚拟节点连向每个树作为新的根， 树上背包一下就好了. md发现tarjan敲错狂wa不止身败名裂.

　　先给出O(nm^2)的做法.

#include<bits/stdc++.h>
using namespace std;
stack<int> s;
int n, m, num, idx, cnt, ans;
int f[205][505], scc[205], h[205], low[205], dfn[205], w[205], c[205], depend[205], cc[205], ww[205], in[205];
struct edge {
    int nxt, v;
}e[405];
inline void add(int u, int v) {
    e[++ num].v = v, e[num].nxt = h[u], h[u] = num;
}
void tarjan(int u) {
    s.push(u);
    dfn[u] = low[u] = ++ idx;
    for (int i = h[u]; i; i = e[i].nxt) {
        int v = e[i].v;
        if (!dfn[v]) tarjan(v), low[u] = min(low[u], low[v]);
        else if (!scc[v]) low[u] = min(low[u], dfn[v]); 
    }
    if (low[u] == dfn[u]) {
        ++ cnt;
        while (true) {
            int x = s.top(); s.pop();
            scc[x] = cnt;
            cc[cnt] += c[x], ww[cnt] += w[x];
            if (x == u) break;
        }
    }
}
void dfs(int u) {
    f[u][ww[u]] = cc[u];
    for (int p = h[u]; p; p = e[p].nxt) {
        int v = e[p].v;
        dfs(v);
        for (int i = m; i >= ww[u]; -- i)
            for (int j = 1; j <= i - ww[u]; ++ j)
                f[u][i] = max(f[u][i], f[u][i - j] + f[v][j]);
    }
}
int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; ++ i) scanf("%d", &w[i]);
    for (int i = 1; i <= n; ++ i) scanf("%d", &c[i]);
    for (int i = 1; i <= n; ++ i) {
        scanf("%d", &depend[i]);
        if (depend[i]) add(depend[i], i);
    }
    for (int i = 1; i <= n; ++ i)
        if (!dfn[i]) tarjan(i);
    num = 0, memset(h, 0, sizeof (h));
    for (int i = 1; i <= n; ++ i)
        if (depend[i] && scc[depend[i]] != scc[i])
            add(scc[depend[i]], scc[i]), in[scc[i]] = 1;
    for (int i = 1; i <= n; ++ i) if (!in[i]) add(0, i);
    dfs(0);
    for(int i = 0; i <= m; i++) ans = max(ans, f[0][i]);
    printf("%d\n", ans);
}

　　就是要有追求更优复杂度的算法的精神~~ 给出O(nm)的算法.

#include<bits/stdc++.h>
using namespace std;
stack<int> s;
int n, m, num, idx, cnt, ans;
int f[205][505], scc[205], h[205], low[205], dfn[205], w[205], c[205], depend[205], cc[205], ww[205], in[205];
struct edge {
    int nxt, v;
}e[405];
inline void add(int u, int v) {
    e[++ num].v = v, e[num].nxt = h[u], h[u] = num;
}
void tarjan(int u) {
    s.push(u);
    dfn[u] = low[u] = ++ idx;
    for (int i = h[u]; i; i = e[i].nxt) {
        int v = e[i].v;
        if (!dfn[v]) tarjan(v), low[u] = min(low[u], low[v]);
        else if (!scc[v]) low[u] = min(low[u], dfn[v]); 
    }
    if (low[u] == dfn[u]) {
        ++ cnt;
        while (true) {
            int x = s.top(); s.pop();
            scc[x] = cnt;
            cc[cnt] += c[x], ww[cnt] += w[x];
            if (x == u) break;
        }
    }
}
void dfs(int u, int V) {
    if (V <= 0) return;
    for (int p = h[u]; p; p = e[p].nxt) {
        int v = e[p].v;
        for (int i = 1; i <= V; ++ i) f[v][i] = f[u][i];
        dfs(v, V - ww[v]);
        for (int i = ww[v]; i <= V; ++ i) f[u][i] = max(f[u][i], f[v][i - ww[v]] + cc[v]);
    }
}
int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; ++ i) scanf("%d", &w[i]);
    for (int i = 1; i <= n; ++ i) scanf("%d", &c[i]);
    for (int i = 1; i <= n; ++ i) {
        scanf("%d", &depend[i]);
        if (depend[i]) add(depend[i], i);
    }
    for (int i = 1; i <= n; ++ i)
        if (!dfn[i]) tarjan(i);
    num = 0, memset(h, 0, sizeof (h));
    for (int i = 1; i <= n; ++ i)
        if (depend[i] && scc[depend[i]] != scc[i])
            add(scc[depend[i]], scc[i]), in[scc[i]] = 1;
    for (int i = 1; i <= n; ++ i) if (!in[i]) add(0, i);
    dfs(0, m);
    for(int i = 0; i <= m; i++) ans = max(ans, f[0][i]);
    printf("%d\n", ans);
}