[JXOI2018模擬] way 樹形dp+啓發式合併

題目大意:給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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章