hdu 1561 The more, The Better(树形dp入门)

http://acm.hdu.edu.cn/showproblem.php?pid=1561

树形dp:

#include<iostream>
#include<cstring>
#include<vector>
using namespace std;

const int mx=222;
int dp[mx][mx];
vector<int>g[mx];

void dfs(int x,int m)
{
    if (m==0) return ;
    for (int i=0;i<g[x].size();i++)
    {
        int cut=g[x][i];
        dfs(cut,m-1);
        for (int j=m;j>=1;j--)
        {
            int v=j;
            for(int k=1;k<v;k++)
            {
                dp[x][v]=max(dp[x][v],dp[x][v-k]+dp[cut][k]);
            }
        }
    }
}

int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        memset(dp,0,sizeof(dp));
        if (!n&&!m) return 0;
        int a,i;
        for (i=0;i<=n;i++) g[i].clear();
        for (i=1;i<=n;i++)
        {
            cin>>a>>dp[i][1];
            g[a].push_back(i);
        }
        dfs(0,m+1);
        cout<<dp[0][m+1]<<endl;
    }
}