题目大意:求遍历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;
}