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

题目描述

传送门

题目大意：在一个地图上，有N座城堡，每座城堡都有一定的宝物，攻克M座城堡使获得的宝物数量尽可能大，有些城堡不能直接攻克，要攻克这些城堡必须先攻克其他某一个特定的城堡。

题解

攻克关系形成的实际上是一片森林
我们将所有的树用0号节点连接，然后将m+1，正常做树形依赖DP就可以了。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 403
using namespace std;
int tot,n,m,nxt[N],point[N],v[N],dp[N][N],val[N];
void add(int x,int y)
{
    tot++; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
}
void dfs(int x,int fa)
{
    for (int i=1;i<=m;i++) dp[x][i]=val[x];
    for (int i=point[x];i;i=nxt[i]){
        if (v[i]==fa) continue;
        dfs(v[i],x);
        for (int j=m;j>=1;j--)
         for (int k=1;k+j<=m;k++)
          dp[x][j+k]=max(dp[x][j+k],dp[x][j]+dp[v[i]][k]);
    }
}
int main()
{
    freopen("a.in","r",stdin);
    while (true) {
        scanf("%d%d",&n,&m);
        if (!n&&!m) break;
        tot=0;
        memset(point,0,sizeof(point));
        memset(dp,0,sizeof(dp));
        for (int i=1;i<=n;i++) {
            int x; scanf("%d%d",&x,&val[i]);
            add(x,i);
        }
        m++;
        dfs(0,0);
        printf("%d\n",dp[0][m]);
    }
}