ZOJ 3201 Tree of Tree（树形DP）

转载请注明出处：忆梦http://blog.csdn.net/yimeng2013/article/details/10274993

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3201

题目大意：先给出N，K，N代表有N个结点，K代表选择K个结点大小的子树，求出最大权值

题解：典型的树形dp，用dp[i][j]:表示以i为根，有j个节点（切树根必须选，所以dp[i][1] = val[i]）,选了该子树中j个节点时候的最大权值。       因为最多选m个，那么这m个中去枚举这个孩子哄占多少个，树根占多少个      dp[root][j] = max( dp[root][j] , dp[root][k] + dp[u][j-k] );   

存储结构用的是链式前向星，A了 写了一个vector版  但不知道为嘛一直WA，以后再补上来

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;

#define N 105
struct Edge
{
    int vert , next;
}e[N * 2];

int dp[N][N];
int head[N];
int val[N];
int vis[N];
int n,m;

void DFS( int root )
{

    int i , j , k , u;

     dp[root][1] = val[root];
    
    for(i=head[root];  i != -1;  i=e[i].next)
    {
         u = e[i].vert;
         if( vis[u] )  
			 continue;
         vis[u]++;

         DFS(u);
         
         for( j=m;  j>=1;  j--)
             for(k=1;  k<=j;  k++)
                 dp[root][j] = max( dp[root][j] , dp[root][k] + dp[u][j-k] );
    }
    
}

void init()
{
	int u, v;
	int num = 1;
	for(int i=1;  i<n;  i++)
      {
         scanf("%d%d" , &u , &v );   
         e[num].vert = v;
         e[num].next = head[u];
         head[u] = num++;
            
         e[num].vert = u;
         e[num].next = head[v];
         head[v] = num++;    
        
       }
}
int main()
{
    int i , j;
    
    while( scanf("%d%d" , &n , &m ) != EOF )
    {
        for(i=0;  i<n;  i++  )
            scanf("%d" , &val[i] );
        
        memset(head , -1 , sizeof(head) );
        memset(vis , 0 , sizeof(vis) );
        memset(dp , 0,  sizeof(dp) );
        
	    init();
        
        vis[0]++;
        DFS(0);
        
		int ans = 0;
        for(i = 0;  i<n;  i++  )
            if( ans < dp[i][m] )
                 ans = dp[i][m] ;

         
         printf("%d\n" , ans );
        
    }
    return 0;
}