hihoCoder1055.刷油漆——树形Dp+01背包

http://hihocoder.com/problemset/problem/1055

一棵有根树，包含根节点1，选择M个连续的节点，使得权值最大

dp[u][j] 表示以i为根的子树中，选择包含根节点的j个连续节点所能获得的最大权值

枚举子节点选择的个数，儿子节点选择的个数，当做01背包

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <algorithm>
#define debug puts("--------");
#define debug1 system("pause");
#define clr(a,b) memset(a,b,sizeof(a))
#define pk push_back
const int maxn = 110;
using namespace std;
int N,M;
int dp[maxn][maxn];
vector<int> g[maxn];
int a[maxn];
void solve(int u,int pre)
{
    dp[u][1]=a[u];
    for(int i=0;i<g[u].size();++i){
        int v=g[u][i];
        if(v==pre) continue;
        solve(v,u);
        for(int i=M;i>=1;--i){//逆序枚举容量，01背包
            for(int j=1;j<i;++j){
                dp[u][i]=max(dp[u][i],dp[u][i-j]+dp[v][j]);
            }
        }
    }
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.txt","r",stdin);
freopen("out.txt","w",stdout);
#endif // ONLINE_JUDGE
    int u,v;
    while(scanf("%d%d",&N,&M)==2){
        for(int i=1;i<=N;++i) g[i].clear();
        clr(dp,0);
        for(int i=1;i<=N;++i) scanf("%d",&a[i]);
        for(int i=0;i<N-1;++i){
            scanf("%d%d",&u,&v);
            g[u].pk(v);
            g[v].pk(u);
        }
        solve(1,-1);
        printf("%d\n",dp[1][M]);
    }
    return 0;
}