HDU 1561 The more, The Better(树形DP）

HDU 1561 The more, The Better(树形DP）

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
#define ll long long
const int maxn = 2e2+5;
const int INF = 0x3f3f3f3f;
///dp[i][j] 表示以i为跟节点的树，选其中的j个节点可以达到的最大值
int dp[maxn][maxn];
vector<int> v[maxn];
int N, M, x, a[maxn];

void dfs(int n, int m) {
    dp[n][1] = a[n];

    int len = v[n].size();
    for(int i = 0; i < len; ++i) {
        if(m > 1)//树形撸
            dfs(v[n][i], m-1);

        for(int j = m-1; j >= 1; j--) {
            for(int k = 1; k <= j; ++k) {
                dp[n][j+1] = max(dp[n][j+1], dp[n][k]+dp[v[n][i]][j+1-k]);//儿子k个节点,父亲j+1-k个节点
            }
        }

    }
}

int main() {
    //freopen("data.in", "r", stdin);
    while(scanf("%d%d", &N, &M) && N+M) {
        memset(dp, 0, sizeof(dp));
        memset(a, 0, sizeof(a));
        for(int i = 0; i <= N; i++)
            v[i].clear();

        for(int i = 1; i <= N; i++) {
            scanf("%d%d", &x, &a[i]);
            v[x].push_back(i);
        }

        dfs(0,M+1); ///0是补的根节点,M+1是可以取几个结点,因为多加了一个0结点 值为0 所以要M+1
        printf("%d\n", dp[0][M+1]);
    }
    return 0;
}