每日一题 POJ 2486 树形dp 树上背包

//AC代码 耗时188MS
#include <iostream>
#include<cstring>
#define MAXN 205
using namespace std;
struct EDGE
{
    int to,next;
} edge[MAXN];
int head[MAXN],ptr;
int val[MAXN];
int N,K;
void add_edge(int u,int v)
{
    edge[++ptr].to=v;
    edge[ptr].next=head[u];
    head[u]=ptr;
}
void add(int u,int v)
{
    add_edge(u,v);
    add_edge(v,u);
}
int dpr[105][205],dpnr[105][205];
void dfs(int now,int f)
{
    dpr[now][0]=dpnr[now][0]=val[now];
    for(int p=head[now]; p; p=edge[p].next)
    {
        int to=edge[p].to;
        if(to==f)continue;
        dfs(to,now);
        for(int k=K; k>=1; k--)
        {
            for(int t=2; t<=k; t++)
                dpnr[now][k]=max(dpnr[now][k],dpr[to][t-2]+dpnr[now][k-t]),
                dpr[now][k]=max(dpr[now][k],dpr[to][t-2]+dpr[now][k-t]);
            for(int t=1; t<=k; t++)
                dpnr[now][k]=max(dpnr[now][k],dpnr[to][t-1]+dpr[now][k-t]);
        }
    }
}
void solve()
{
    memset(dpr,0xcf,sizeof(dpr));
    memset(dpnr,0xcf,sizeof(dpnr));
    dfs(1,1);
    int ans=0;
    for(int i=0; i<=K; i++)
        ans=max(ans,dpnr[1][i]);
    cout<<ans<<endl;
    memset(edge,0,sizeof(edge));
    memset(head,0,sizeof(head));
    ptr=0;
}
int main()
{
    while(cin>>N>>K)
    {
        for(int i=1; i<=N; i++)
            cin>>val[i];
        for(int i=1; i<N; i++)
        {
            int a,b; cin>>a>>b;
            add(a,b);
        }
        solve();
    }
}