ZOJ3201 Tree of Tree 树状dp入门

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3201

题目大意：给出一个树，每个点有一个值，选择k个点要求值最大，k各点连成一颗子树，求最大值

代码：

#include <math.h>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define MAXN 105
#define INF 0x7fffffff
#define eps 1e-10
#define MAX(x,y) ((x)>(y)? (x):(y))
#define MIN(x,y) ((x)<(y)? (x):(y))
#define MEM(a) (memset((a),0,sizeof(a)))
#define FRE freopen("input.txt","r",stdin)
using namespace std;

int n,m;
int val[MAXN],head[MAXN],vis[MAXN],dp[MAXN][MAXN];
struct edge
{
    int u,v,next;
}e[2*MAXN];


void dfs(int u)
{
    vis[u]=1;
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        int v=e[i].v;
        if(!vis[v])
        {
            dfs(v);
            for(int j=m;j>0;j--)
                for(int k=0;k<=m-j;k++)
                    dp[u][j+k]=MAX(dp[u][j+k],dp[u][j]+dp[v][k]);
        }
    }
    return;
}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(head,-1,sizeof(head));
        MEM(dp);
        for(int i=0;i<n;i++)
            scanf("%d",&val[i]);
        int cnt=0;
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            e[cnt].u=u; e[cnt].v=v; e[cnt].next=head[u]; head[u]=cnt++;
            e[cnt].u=v; e[cnt].v=u; e[cnt].next=head[v]; head[v]=cnt++;
        }
        for(int i=0;i<n;i++)
            dp[i][1]=val[i];
        int ans=-1;
        MEM(vis);
        dfs(0);
        for(int i=0;i<n;i++)
        {
            ans=MAX(ans,dp[i][m]);
        }
        printf("%d\n",ans);
    }
    return 0;
}