hdu4616 树形DP

多校第二场了，被打击的遍体鳞伤啊，比赛结束的时候看了这题，一眼就想到树形DP，真正写起来才发现原来这么麻烦，状态合并有一部分要去局部最优，只好加了一维状态记录当前节点往下i次trap且以trap值为1的节点结尾的最优值。题目不算很难，细节调起来很纠结。

ACcode：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int NS=50010;

struct node{
    int val,cot;
}a[NS];

int N,C,top,ans;
int head[NS],next[NS<<1],to[NS<<1],vis[NS];

void add(int u,int v)
{
    to[top]=v;
    next[top]=head[u];
    head[u]=top++;

    to[top]=u;
    next[top]=head[v];
    head[v]=top++;
}

int up[5][2];

void dfs(int rt)
{
    int dp[5][2]={0},c=a[rt].cot,v=a[rt].val;
    ans=max(ans,dp[c][0]=v);
    for (int i=head[rt];i!=-1;i=next[i])
    {
        int son=to[i];
        if (!vis[son])
        {
            vis[son]=1,dfs(son);
            for (int j=0;j<=C;j++)
            for (int k=0;j+k<=C;k++)
            {
                if (j+k==C)
                ans=max(ans,dp[j][0]+up[k][1]),
                ans=max(ans,dp[j][1]+up[k][0]);
                else
                ans=max(ans,dp[j][0]+up[k][0]);
            }

            for (int j=c;j<=C;j++)
                dp[j][0]=max(dp[j][0],up[j-c][0]+v),
                dp[j][1]=max(dp[j][1],up[j-c][1]+v);
        }
    }

    for (int i=0;i<C;i++)
    ans=max(ans,dp[i][0]);

    if (c)
    ans=max(ans,dp[C][0]),
    dp[C][0]=dp[C][1],
    dp[1][1]=dp[c][1]=v;
    else
    ans=max(ans,dp[C][1]);

    for (int i=0;i<=C;i++)
    up[i][0]=dp[i][0],
    up[i][1]=dp[i][1];
    up[0][1]=0;
}

int main()
{
    int T,u,v;
    for (scanf("%d",&T);T;T--)
    {
        top=ans=0;
        memset(vis,0,sizeof(vis));
        memset(head,-1,sizeof(head));
        scanf("%d %d",&N,&C);
        for (int i=0;i<N;i++)
            scanf("%d %d",&a[i].val,&a[i].cot);
        for (int i=1;i<N;i++)
            scanf("%d %d",&u,&v),add(u,v);
        vis[0]=1,dfs(0);
        printf("%d\n",ans);
    }
    return 0;
}