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 403using 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]);    }}