hdu 1561 The more, The Better（树形依赖背包）

The more, The Better

Time Limit: 6000/2000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 10384    Accepted Submission(s): 6042

 

Problem Description

ACboy很喜欢玩一种战略游戏，在一个地图上，有N座城堡，每座城堡都有一定的宝物，在每次游戏中ACboy允许攻克M个城堡并获得里面的宝物。但由于地理位置原因，有些城堡不能直接攻克，要攻克这些城堡必须先攻克其他某一个特定的城堡。你能帮ACboy算出要获得尽量多的宝物应该攻克哪M个城堡吗？

 

 

Input

每个测试实例首先包括2个整数，N,M.(1 <= M <= N <= 200);在接下来的N行里，每行包括2个整数，a,b. 在第 i 行，a 代表要攻克第 i 个城堡必须先攻克第 a 个城堡，如果 a = 0 则代表可以直接攻克第 i 个城堡。b 代表第 i 个城堡的宝物数量, b >= 0。当N = 0, M = 0输入结束。

 

 

Output

对于每个测试实例，输出一个整数，代表ACboy攻克M个城堡所获得的最多宝物的数量。

 

 

Sample Input

3 2 0 1 0 2 0 3 7 4 2 2 0 1 0 4 2 1 7 1 7 6 2 2 0 0

 

 

Sample Output

5 13

 

 

Author

8600

 

 

Source

HDU 2006-12 Programming Contest

 

 

Recommend

LL   |   We have carefully selected several similar problems for you:  1011 2159 2639 1203 1712 

 

Statistic | Submit | Discuss | Note

Home | Top

Hangzhou Dianzi University Online Judge 3.0 Copyright © 2005-2019 HDU ACM Team. All Rights Reserved. Designer & Developer : Wang Rongtao LinLe GaoJie GanLu Total 0.024001(s) query 5, Server time : 2019-04-04 08:11:07, Gzip enabled

Administration

 

显然通过树上的关系像是一个依赖关系，并且是森林中求依赖关系，并且取m个节点要求权值最大

首先我可以找一个0节点，把本来不联通的森林连起来，然后变为一颗树，接着就是树形依赖背包

的过程了，
f[i][j][k] 代表以i节点的子树下，i节点前k个孩子，干点j个城堡最多可以获得多少价值，然后考虑状

态转移：当u是v的父节点的时候，转移就是01背包只不过在儿子和父亲节点上做转移 

    int m = size;
    for(int i = 2;i <= m;i++){
        for(int j = 1;j <= i;j++){
            f[u][i][j] = max(f[u][j][k],f[u][i - j][i - 1] + f[v][j][i];
        }
    }

会发现n^3会爆空间，然后考虑01背包的降维，我们可以把前k个孩子的维度去掉，

f[i][j]表示以i节点为根的子树下，干掉j个城堡可以获得的最大价值，那我们的ans就是f[0][m + 1]

即从0为根节点，m+1容量的，因为我们选了根节点0,0的权值是0的

接着考虑转移，

        for(int j = m;j > 1;j--){
            for(int i = 0;i < j;i++)
            f[x][j] = max(f[x][j],f[x][j - i] + f[d][i]);
        }

 

代码如下：

#include <bits/stdc++.h>
#include <time.h>
#define fi first
#define se second
#define endll "\n"
using namespace std;

typedef long long ll;
typedef double db;
int xx[4] = {1,-1,0,0};
int yy[4] = {0,0,1,-1};
const double eps = 1e-9;
typedef pair<char,int>  P;
const int maxn = 2e6 + 5000;
const ll mod = 1e9 + 7;
inline int sign(db a) { return a < -eps ? -1 : a > eps;}
inline int cmp(db a,db b){ return sign(a - b);}
ll mul(ll a,ll b,ll c) { ll res = 1; while(b) {  if(b & 1) res *= a,res %= c;  a *= a,a %= c,b >>= 1;  }  return res;}
ll phi(ll x) {  ll res = x;  for(ll i = 2; i * i <= x; i++) { if(x % i == 0) res = res / i * (i - 1);   while(x % i == 0) x /= i;   }  if(x > 1) res = res / x  * (x - 1);    return res;}
int fa[maxn];
int Find(int x) {   if(x != fa[x]) return fa[x] = Find(fa[x]);  return fa[x];}
vector<vector<int> >v;
int val[maxn];
int f[500][500];///代表i节点的子树下，攻击j个节点能获得的最多价值 

/**
    int m = size;
    for(int i = 2;i <= m;i++){
        for(int j = 1;j <= i;j++){
            f[u][i][j] = max(f[u][j][k],f[u][i - j][i - 1] + f[v][j][i];
        }
    }
**/ 
void dfs(int x,int m){
    f[x][1] = val[x];
    for(auto d:v[x]){
        dfs(d,m);
        for(int j = m;j > 1;j--){
            for(int i = 0;i < j;i++)
            f[x][j] = max(f[x][j],f[x][j - i] + f[d][i]);
        }
    }
}
ll c,n,k,w,q,m;
int main() {
    ios::sync_with_stdio(false);
    while(cin >> n >> m){
        if(n == 0 &&  m == 0) break;
        v.clear();
        v.resize(n + 500);
        memset(f,0,sizeof(f));
        for(int i = 1; i<= n;i++){
            int a;
            cin >> a >> val[i];
            v[a].push_back(i);
        }
        dfs(0,m + 1);
        cout << f[0][m + 1] << endll;
    }
//    cerr << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
    return 0;
}
//
//7 4
//1 2 2
//2 0 1
//3 0 4
//4 2 1
//5 7 1
//6 7 6
//7 2 2