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);
}
}