ZOJ 3201 树形DP

题目链接

题意：

一颗树，树上各点有权值。要求从中选取一个权值和最大且节点数等于k的子树，输出最大权值和。

思路：

定义dp[i][j]表示从以i为根的子树中选取j个节点可以获得的最大值。

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

因为是子树，所以需要在每个节点处ans = max ( ans , dp[n][m] )

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,cost;
}es[maxm];

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

int ans,dp[maxn][maxn],s[maxn];

void dfs( int u , int f )
{
    for ( int i=1 ; i<=n ; i++ )
        dp[u][i] = s[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=n ; j>=1 ; j-- )
                for ( int k=1 ; j+k<=n ; k++ )
                    dp[u][j+k] = max( dp[u][j+k] , dp[u][j]+dp[v][k] );
        }
    }
    ans = max ( ans , dp[u][m] );
}

int main()
{
    while ( scanf ( "%d%d" , &n , &m )==2 )
    {
        for ( int i=0 ; i<n ; i++ )
            scanf ( "%d" , &s[i] );
        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 );
        }
        ans = 0;
        dfs( 0 , -1 );
        printf( "%d\n" , ans );
    }
    return 0;
}