HDU 1011 树形DP

题目链接

题意：

一颗有根树，各点有一个消耗值和价值（即消耗消耗值即可获得对应价值），并且若想前往某点必须先去过其父节点，若想进入某点必须当前剩余可以消耗的东西>0。

思路：

定义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 = 110;
const int maxm = 210;

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

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

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

void dfs( int u , int f )
{
    for ( int i=0 ; i<=m ; i++ )
        if ( i<b[u] ) dp[u][i] = 0;
        else dp[u][i] = c[u];
    for ( int i=head[u] ; i!=-1 ; i=es[i].next )
    {
        int v = es[i].to;
        if ( v!=f )
        {
            dfs( v , u );
            for ( int j=m ; j>=b[u] ; j-- )
                for ( int k=1 ; j+k<=m ; 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==-1&&m==-1 ) break;
        for ( int i=1 ; i<=n ; i++ )
        {
            scanf ( "%d%d" , &b[i] , &c[i] );
            if ( b[i]%20==0 ) b[i] = b[i]/20;
            else b[i] = b[i]/20+1;
        }
        tol = 0;
        memset ( head , -1 , sizeof(head) );
        for ( int i=1 ; i< n ; i++ )
        {
            int u,v;
            scanf ( "%d%d" , &u , &v );
            addedge ( u , v );
            addedge ( v , u );
        }
        if ( !m )
        {
            printf ( "0\n" );
            continue;
        }
        dfs( 1 , 0 );
        printf ( "%d\n" , dp[1][m] );
    }
    return 0;
}