hdu 1561 The more, The Better 树形dp

题目

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1561

题目来源：http://acm.hust.edu.cn/vjudge/contest/view.action?cid=105318#problem/H

简要题意：中文不表。

题解

建有向图，从$a$指向$i$，然后只要从$0$开始dfs即可。

$dp[i][j]$表示选定$i$，$i$及子树选定共$j$个点的最大值。

转移起来就是像个背包一样去转移，注意这样要反向来转移。

对于每个儿子，将答案合并到自己的答案里头，枚举两边有多少点。

注意加上虚拟的$0$号点后总共有$m+1$个点。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>

#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef vector<int> VI;
typedef pair<int,int> PII;
LL powmod(LL a,LL b, LL MOD) {LL res=1;a%=MOD;for(;b;b>>=1){if(b&1)res=res*a%MOD;a=a*a%MOD;}return res;}
// head
const int N = 205;

struct Edge {
    int to, nxt;
    Edge(int to, int nxt) : to(to), nxt(nxt) {}
    Edge() {}
};

int head[N];
Edge e[N];

void addEdge(int from, int to, int cnt) {
    e[cnt] = Edge(to, head[from]);
    head[from] = cnt;
}

void init(int n) {
    for (int i = 0; i <= n; i++) head[i] = -1;
}

int a[N];
int dp[N][N];
int n, m, v;

void dfs(int x) {
    dp[x][1] = a[x];
    for (int i = head[x]; ~i; i = e[i].nxt) {
        int to = e[i].to;
        dfs(to);
        for (int j = m; j > 0; j--) {
            for (int k = 1; k < j; k++) {
                dp[x][j] = max(dp[x][j], dp[to][j-k] + dp[x][k]);
            }
        }
    }
}

int main() {
    while (scanf("%d%d", &n, &m) == 2 && n) {
        init(n);
        m++;
        int ec = 0;
        for (int i = 1; i <= n; i++) {
            scanf("%d%d", &v, a+i);
            addEdge(v, i, ec++);
        }
        dfs(0);
        printf("%d\n", dp[0][m]);
        memset(dp, 0, sizeof dp);
    }
    return 0;
}