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 ;
 }