题意:求用最多用k步采摘最多的苹果数,点u->v算一步,点v->u即回来也算一步 、
参考http://blog.csdn.net/xingyeyongheng/article/details/17994781
应该是挺经典的了。。我也不知道怎么处理走回来, 的,
没想到是再开一维。。
int val[N];
int dp[N][210][2];
int fst[N],nxt[2*N],vv[N],tot;
int n,k;
void add(int u,int v){
vv[tot]=v;nxt[tot]=fst[u];fst[u]=tot++;
}
void dfs(int u,int fa){
for(int i=fst[u];~i;i=nxt[i]){
int v=vv[i];
if(v==fa)continue;
dfs(v,u);
for(int j=k;j>=0;--j){
for(int t=1;j+t<=k;++t){
dp[u][j+t][0]=max(dp[u][j+t][0],dp[u][j][1]+dp[v][t-1][0]+val[v]);
if(t>=2)dp[u][j+t][0]=max(dp[u][j+t][0],dp[u][j][0]+dp[v][t-2][1]+val[v]);
if(t>=2)dp[u][j+t][1]=max(dp[u][j+t][1],dp[u][j][1]+dp[v][t-2][1]+val[v]);
}
}
}
}
int main(){
//freopen("in.txt","r",stdin);
while(~sf("%d%d",&n,&k)){
mem(dp,0);mem(fst,-1);tot=0;
rep(i,1,n)sf("%d",&val[i]);
rep(i,1,n-1){ int u,v;sf("%d%d",&u,&v);add(u,v);add(v,u); }
dfs(1,1);
pf("%d\n",dp[1][k][0]+val[1]);
}
}