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

题面

题意：有n个城堡形成一棵树，如果要攻克某个节点必须要先攻克其父节点。每个节点有i个宝物，最多能攻克m个城堡。问攻克哪些城堡能获得最多的宝物数量。

思路:

首先本题有两个限制：

1.必须先攻打父节点然后才能攻打子节点

2.最多只能攻克m个城堡

攻占城堡的顺序是自上而下的，但我们可以从叶子节点自下而上不断更新节点的最优值求子树的最优解，最终得到根的最优解。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<stack>
#include<map>
using namespace std;
const int N=500;
struct E
{
    int val;
    int next;
}edge[N];
int cas[N],vis[N],c,dp[N][N],head[N];
void add(int x,int y)
{
    edge[c].val=y;
    edge[c].next=head[x];
    head[x]=c;
    c+=1;
}
int selMax(int x,int y)
{
    return x>=y?x:y;
}
void DFS(int root,int M)
{
    vis[root]=1;
    dp[root][1]=cas[root];
    for(int i=head[root];i!=-1;i=edge[i].next)
    {
        int cur=edge[i].val;
        if(!vis[cur])
            DFS(cur,M-1);
        for(int x=M;x>=1;x--)
        {
            for(int y=1;y<x;y+=1)
            {
                dp[root][x]=selMax(dp[root][x],dp[root][x-y]+dp[cur][y]);
            }
        }
    }
}
int main()
{
    int n,M;
    while(~scanf("%d%d",&n,&M))
    {
        if(n+M==0) break;
        memset(vis,0,sizeof(vis));
        memset(head,-1,sizeof(head));
        memset(dp,0,sizeof(dp));
        c=0;
        for(int i=1;i<=n;i+=1)
        {
            int x;
            scanf("%d%d",&x,&cas[i]);
            add(x,i);
        }
        DFS(0,M+1);
        printf("%d\n",dp[0][M+1]);
    }
    return 0;
}