多校第二场了,被打击的遍体鳞伤啊,比赛结束的时候看了这题,一眼就想到树形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;
}