poj2486 Apple Tree 树上分组背包

题目大意:求遍历k个几点权值和的最大值 (n<=200) 空间限制65536K
扶额 这么一算n^3空间是可以的啊……怎么会MLE……
好吧 改成二维的 用三维的写题解 好理解
dp[i][j][0][cnt] : i的子树走j步回到i用前cnt个分组(son)的最大权值
dp[i][j][1][cnt] : i的子树走j步不回到i用前cnt个分组(son)的最大权值
dp[i][j][0][cnt]考虑前cnt-1个点回来了且这个带你回来了dp[i][j][1][cnt]考虑成前cnt-1个点回来了第cnt个点不回来 或者 前cnt-1个点有没回来的第cnt个点回来的
还有些细节 比如 dp[i][j][0][cnt] 初始之类的
转移方程详见代码
丑的一笔

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=205;
struct E{int to,nxt;}edge[N*2];
int dp[N][N][2];
int son[N],val[N];
int idx[N],tot;
int h,n,T;
void init(){
    tot=1;
    memset(edge,0,sizeof(E));
    memset(idx,0,sizeof(idx));
    memset(dp,0,sizeof(dp));
    memset(son,0,sizeof(son));
    memset(val,0,sizeof(val));
}
void addedge(int from,int to){
    edge[tot].to=to;edge[tot].nxt=idx[from];idx[from]=tot++;
}
void dfs(int x,int fa){
    for(int t=idx[x];t;t=edge[t].nxt){
        E e=edge[t];
        if(e.to!=fa) {
            dfs(e.to,x);son[x]++;
            for(int j=h;j>=1;j--){
                dp[x][j][0]=max(dp[x][j-1][0],dp[x][j][0]);
                dp[x][j][1]=max(dp[x][j-1][1],dp[x][j][1]);
                for(int k=1;k<=j;k++){
                    if(k>=2) 
                        dp[x][j][0]=max(dp[x][j][0],dp[x][j-k][0]+dp[e.to][k-2][0]);
                    if(k>=2) 
                        dp[x][j][1]=max(dp[x][j][1],dp[x][j-k][1]+dp[e.to][k-2][0]);
                    if(k>=1) 
                        dp[x][j][1]=max(dp[x][j][1],dp[x][j-k][0]+max(dp[e.to][k-1][0],dp[e.to][k-1][1]));
                }
            }
        }
    }
}
int main(){
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
    while(~scanf("%d%d",&n,&h)){
        init();
        int x,y;
        for(int i=1;i<=n;i++) {
            scanf("%d",&val[i]);
            for(int j=0;j<=h;j++) dp[i][j][0]=val[i];
            for(int j=1;j<=h;j++) dp[i][j][1]=val[i];
        }
        for(int i=1;i<n;i++) scanf("%d%d",&x,&y),addedge(x,y),addedge(y,x);
        dfs(1,-1);
        printf("%d\n",max(dp[1][h][1],dp[1][h][0]));
    }
    return 0;
}