hdu4616 树形dp，分治思想

刚做这题的时候感觉很麻烦，而别人却说是经典题，当时我一点儿没看出来经典。。。知道今天看了漆子超的论文，里面第一个例题就和这个差不多。

先是分治思想，然后对于每个根节点，大概想像成向下拉了许多条链出来就行了，有个小技巧可以将时间复杂度降低。

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int maxn=50005;
const int inf=0x3fffffff;
ll val[maxn];int trap[maxn];
ll ans,n,cc;
ll dp[maxn][5][2];
struct edge
{
    int to,next;
}ee[maxn*2];
int e[maxn],ecnt;
void addedge(int u,int v)
{
    ee[ecnt].to=v;ee[ecnt].next=e[u];e[u]=ecnt++;
    ee[ecnt].to=u;ee[ecnt].next=e[v];e[v]=ecnt++;
}
void init()
{
    int i,j,k;
    for(i=1;i<=n;++i)
    {
        for(j=0;j<=cc;++j)
        {
            for(k=0;k<2;++k)
            {
                dp[i][j][k]=-inf;
            }
        }
    }
}
void dfs(int f,int u)
{
    int v,i,j,k;
    dp[u][trap[u]][trap[u]]=val[u];
    for(i=e[u];i!=-1;i=ee[i].next)
    {
        v=ee[i].to;
        if(v==f)
            continue;
        dfs(u,v);
        for(j=0;j<=cc;++j)
        {
            for(k=0;j+k<=cc;++k)
            {
                if(j+k<cc)
                {
                    ans=max(ans,dp[u][j][0]+dp[v][k][0]);
                }
                if(j+k<=cc)
                {
                    ans=max(ans,dp[u][j][1]+dp[v][k][1]);
                }
                if(j!=cc)
                {
                    ans=max(ans,dp[u][j][0]+dp[v][k][1]);
                }
                if(k!=cc)
                {
                    ans=max(ans,dp[u][j][1]+dp[v][k][0]);
                }
            }
        }
        for(j=0;j<cc;++j)
        {
            dp[u][j+trap[u]][0]=max(dp[u][j+trap[u]][0],dp[v][j][0]+val[u]);
            dp[u][j+trap[u]][1]=max(dp[u][j+trap[u]][1],dp[v][j][1]+val[u]);
        }
        if(trap[u]==0)
        {
            dp[u][cc][1]=max(dp[u][cc][1],dp[v][cc][1]+val[u]);
        }
    }
}
int main()
{
    int t,i,j,u,v;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&cc);
        for(i=1;i<=n;++i)
        {
            scanf("%d%d",&val[i],&trap[i]);
        }
        memset(e,-1,sizeof(e));ecnt=0;
        for(i=1;i<n;++i)
        {
            scanf("%d%d",&u,&v);
            u++;v++;
            addedge(u,v);
        }
        init();
        ans=-inf;
        dfs(-1,1);
        printf("%d\n",ans);
    }
    return 0;
}