題目大意:給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;
}