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

经典的树形DP。

先将多叉树变成二叉树，然后进行DP。

dp[i][j]表示以i为根，可选j个时候的最优解。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn=205;
int ch[maxn][2];
int f[maxn][maxn];
int v[maxn];
int dp(int a,int b)
{
    if(a==-1||b==0)
        return 0;
    if(f[a][b]!=-1)
        return f[a][b];
    int &ans=f[a][b];
    ans=dp(ch[a][1],b);
    for(int i=1; i<=b; ++i)
        ans=max(dp(ch[a][0],i-1)+v[a]+dp(ch[a][1],b-i),ans);
    return ans;
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(!n&&!m)
            break;
        memset(ch,-1,sizeof(ch));
        memset(f,-1,sizeof(f));
        for(int i=1; i<=n; ++i)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            v[i]=b;
            ch[i][1]=ch[a][0];
            ch[a][0]=i;
        }
        ch[0][1]=ch[0][0];
        ch[0][0]=-1;
        printf("%d\n",dp(0,m));
    }
    return 0;
}