[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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值