HDU 1561 树形DP

题目链接

题意：

有n个城堡你可以攻占m个城堡并获得其中的宝物，但是攻占有些城堡需要先攻占某些特定城堡，请问最多可以获得多少价值的宝物。

思路：

将无需先攻占特定城堡的城堡视为需要先攻占0号城堡（我们自行添加的，城堡中的宝物为0），那么整个城堡地图就变成了一颗n+1个节点的有根树。

定义dp[i][j]表示在城堡i为根的子树里攻占j个城堡的最大收益

状态转移方程dp[u][j+k] = max( dp[u][j+k] , dp[u][j]+dp[v][k] )

C++代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 210;

int n,m,tol,head[maxn];
struct edge
{
    int to,next;
}es[maxn];

void addedge( int u , int v )
{
    es[tol].to = v;
    es[tol].next = head[u];
    head[u] = tol++;
}

int f[maxn],b[maxn],dp[maxn][maxn];

void dfs( int u )
{
    for ( int i=1 ; i<=m+1 ; i++ )
        dp[u][i] = b[u];
    for ( int i=head[u] ; i!=-1 ; i=es[i].next )
    {
        int v = es[i].to;
        dfs( v );
        for ( int j=m+1 ; j>=1 ; j-- )
            for ( int k=1 ; j+k<=m+1 ; k++ )
                dp[u][j+k] = max( dp[u][j+k] , dp[u][j]+dp[v][k] );
    }
}

int main()
{
    while( scanf( "%d%d" , &n , &m )==2 )
    {
        if ( n==0&&m==0 ) break;
        tol = 0;
        memset ( head , -1 , sizeof(head) );
        for ( int i=1 ; i<=n ; i++ )
        {
            scanf ( "%d%d" , &f[i] , &b[i] );
            addedge ( f[i] , i );
        }
        b[0] = 0;
        dfs( 0 );
        printf ( "%d\n" , dp[0][m+1] );
    }
    return 0;
}