hdu 1561(树形dp)

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

思路：dp[u][i]表示以u为根的树选了i个子节点。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define MAXN 222

int n,m;
int dp[MAXN][MAXN];
int val[MAXN];
vector<vector<int> >G;

void dfs(int u,int father)
{
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(v==father)continue;
        dfs(v,u);
        for(int j=m;j>=1;j--){
            for(int k=1;k<=j;k++){
                dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]);
            }
        }
    }
    if(u!=0){
        for(int j=m;j>=1;j--){
            dp[u][j]=dp[u][j-1]+val[u];
        }
    }
}

int main()
{
    int u,w;
    while(~scanf("%d%d",&n,&m)){
        if(n==0&&m==0)break;
        G.clear();
        G.resize(n+2);
        for(int i=1;i<=n;i++){
            scanf("%d%d",&u,&w);
            G[u].push_back(i);
            val[i]=w;
        }
        memset(dp,0,sizeof(dp));
        dfs(0,-1);
        printf("%d\n",dp[0][m]);
    }
    return 0;
}