ZOJ3201 Tree of Tree(树形DP)

题目大概求一棵树中大小为k的子树的最大权和。

dp[u][k]表示以u为根的子树中包含u结点的大小为k的子树的最大权和,然后树上背包转移转移很容易。。

 #include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 111 struct Edge{
int v,next;
}edge[MAXN<<];
int NE,head[MAXN];
void addEdge(int u,int v){
edge[NE].v=v; edge[NE].next=head[u];
head[u]=NE++;
} int n,k,val[MAXN],d[MAXN][MAXN];
void dp(int u,int fa){
d[u][]=; d[u][]=val[u];
for(int i=head[u]; i!=-; i=edge[i].next){
int v=edge[i].v;
if(v==fa) continue;
dp(v,u);
for(int a=k; a>=; --a){
for(int b=; b<a; ++b){
if(d[u][a-b]==- || d[v][b]==-) continue;
d[u][a]=max(d[u][a],d[u][a-b]+d[v][b]);
}
}
}
} int main(){
int a,b;
while(~scanf("%d%d",&n,&k)){
for(int i=; i<n; ++i){
scanf("%d",val+i);
}
NE=;
memset(head,-,sizeof(head));
for(int i=; i<n; ++i){
scanf("%d%d",&a,&b);
addEdge(a,b);
addEdge(b,a);
}
memset(d,-,sizeof(d));
dp(,);
int res=-;
for(int i=; i<n; ++i){
res=max(res,d[i][k]);
}
printf("%d\n",res);
}
return ;
}
上一篇:docker批量删除容器、镜像


下一篇:【SpringCloud微服务实战学习系列】服务治理Spring Cloud Eureka