hdu 4616 Game 多校第二场

树形dp  细节比较多，看了别人博客才会写。。。

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
using namespace std;

typedef __int64 lld;
const int MAXN = 50050;
const int inf= 0x7f7f7f7f;
int dp[MAXN][4][2];//dp[u][c][k] 表示u为根  走过c个trap 方向为k（0表示出，1入） 的最大值
int ans;
int w[MAXN],t[MAXN],vis[MAXN];
int n,c;
vector<int>vv[MAXN];
void dfs(int u)
{
    vis[u]=1;
    dp[u][t[u]][0]=dp[u][t[u]][1]=w[u];
    int siz=vv[u].size();
    for(int i=0;i<siz;i++)
    {
        int v=vv[u][i];
        if(vis[v]) continue;
        dfs(v);
        for(int k=0;k<=c;k++)
            for(int j=0;j+k<=c;j++)
        {
            if(k!=c)
            ans=max(ans,dp[u][k][0]+dp[v][j][1]);//k==c 时u结点不能出
            if(j!=c)
            ans=max(ans,dp[u][k][1]+dp[v][j][0]);//j==c 时v结点不能出
            if(k+j<c)
            ans=max(ans,dp[u][k][0]+dp[v][j][0]);
        }
        for(int j=0;j+t[u]<=c;j++)
            dp[u][j+t[u]][0]=max(dp[u][j+t[u]][0],dp[v][j][0]+w[u]);
        for(int j=1;j+t[u]<=c;j++)
            dp[u][j+t[u]][1]=max(dp[u][j+t[u]][1],dp[v][j][1]+w[u]);//如果j==0，不能到达u 
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&c);
        for(int i=0;i<=n;i++) vv[i].clear();
        memset(vis,0,sizeof(vis));
        memset(dp,0,sizeof(dp));
        for(int i=0;i<n;i++)
            scanf("%d%d",&w[i],&t[i]);
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            vv[u].push_back(v);
            vv[v].push_back(u);
        }
        ans=0;
        dfs(0);
        printf("%d\n",ans);
    }
}