HDU - 1561 The more, The Better（樹形dp）

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

題目意思：中文題。

思路：樹上揹包，當u=0（即在根節點時），往下dp可以不選自己，因爲自己無價值。

其他情況就必須選自己（因爲不選自己就無法選兒子）。

具體看代碼

代碼：

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define eps 1e-8
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
using namespace std;
typedef long long LL;
typedef pair<int, int> P;
const int maxn = 1e4 + 5;
const int mod = 1e9 + 7;

int n,m;
int v[205];
vector<int>G[205];
int dp[205][205];
void dfs(int x){
    for (int i=0;i<G[x].size();i++){
        int to=G[x][i];
        dfs(to);
        if (x==0){
            for (int j=m;j>=0;j--){
                for (int k=0;k<=j;k++){
                    dp[x][j]=max(dp[x][j],dp[x][j-k]+dp[to][k]);
                }
            }
        }
        else {
            for (int j=m;j>=1;j--){
                for (int k=0;k<j;k++){
                    dp[x][j]=max(dp[x][j],dp[x][j-k]+dp[to][k]);
                }
            }
        }
    }
}
int main() {
    while (~scanf ("%d%d",&n,&m)&&n){
        for (int i=0;i<205;i++) G[i].clear();
        memset(dp,0,sizeof(dp));
        for (int i=1;i<=n;i++){
            int fa;
            scanf ("%d%d",&fa,&dp[i][1]);
            G[fa].push_back(i);
        }
        dfs(0);
        printf ("%d\n",dp[0][m]);
    }
    return 0;
}