题目大意:给n个点的有根树,每个点有个权值和大小,q个询问,每个询问为x,s,意为在x子树中,选出大小之和不超过s的点,其中最大权值和是多少。
考场上写个暴力,考后顺便就学了启发式合并。
启发式合并也就是一种思想,即把小的合并到大的上去,可以优化时间复杂度。
每次直接把重儿子的dp数组赋给该点,然后再对轻儿子进行背包
#include<bits/stdc++.h>
#define maxn 5005
using namespace std;
int n,w[maxn],vv[maxn];
struct node{
int to;
node*next;
}*con[maxn];
void addedge(int x,int y)
{
node*p=new node;
p->to=y;
p->next=con[x];
con[x]=p;
}
int hs[maxn],sz[maxn],dfn[maxn],en[maxn],tim;
int dp[maxn][maxn];
void dfs(int v,int fa)
{
sz[v]=1;dfn[v]=++tim;
for(node*p=con[v];p;p=p->next)
if(p->to!=fa){
dfs(p->to,v);
sz[v]+=sz[p->to];
if(sz[p->to]>sz[hs[v]]) hs[v]=p->to;
}
en[v]=tim;
if(hs[v]) for(int i=0;i<=5000;++i) dp[v][i]=dp[hs[v]][i];
for(int i=1;i<=n;++i)
if(dfn[i]>=dfn[v]&&dfn[i]<=en[v]){
if(!hs[v]||dfn[i]<dfn[hs[v]]||dfn[i]>en[hs[v]])
for(int j=5000-vv[i];j>=0;j--)
dp[v][j+vv[i]]=max(dp[v][j+vv[i]],dp[v][j]+w[i]);
}
for(int j=0;j<=5000;++j) dp[v][j+1]=max(dp[v][j+1],dp[v][j]);
}
int main()
{
scanf("%d",&n);
for(int i=1;i<n;++i)
{
int x,y;
scanf("%d%d",&x,&y);
addedge(x,y);addedge(y,x);
}
for(int i=1;i<=n;++i) scanf("%d%d",&w[i],&vv[i]);
dfs(1,0);
int q;
scanf("%d",&q);
while(q--){
int x,y;
scanf("%d%d",&x,&y);
printf("%d\n",dp[x][y]);
}
return 0;
}