[jzoj 5661] 药香沁鼻 解题报告 (DP+dfs序)

interlinkage:

https://jzoj.net/senior/#contest/show/2703/0

description:

solution:

• 注意到这本质就是一个背包，只是选了一个点就必须把它到根节点的所有的点都选上
• 考虑如何转移这个背包，发现一个点要么转移到$dfs$序比它大$1$的点上，要么转移到比这个点子树中$dfs$序最大的点的$dfs$序大$1$的点上
• 前者表示这条链继续选，后者表示放弃这条链
• 易得所有的状态都会被转移到
• 这是一个很经典的问题

code:

#include<algorithm>
#include<cstring>
#include<iostream>
#include<cstdio>
using namespace std;

const int N=5e3+;
const int M=1e4+;
int n,p,ans,tot,tim;
int w[N],f[N],v[N],head[N],dp[N][M],id[N],st[N],ed[N];
struct EDGE
{
    int to,nxt;
}edge[N<<];
inline int read()
{
    char ch=getchar();int s=,f=;
    while (ch<''||ch>'') {if (ch=='-') f=-;ch=getchar();}
    while (ch>=''&&ch<='') {s=(s<<)+(s<<)+ch-'';ch=getchar();}
    return s*f;
}
void add(int u,int v)
{
    edge[++tot]=(EDGE){v,head[u]};
    head[u]=tot;
}
void dfs(int x)
{
    st[x]=++tim;id[tim]=x;
    for (int i=head[x];i;i=edge[i].nxt)
        dfs(edge[i].to);
    ed[x]=tim;
}
int main()
{
    freopen("medicine.in","r",stdin);
    freopen("medicine.out","w",stdout);
    n=read();p=read();
    w[]=read();f[]=read();v[]=read();
    for (int i=;i<=n;i++)
    {
        w[i]=read();f[i]=read();v[i]=read();
        add(f[i],i);
    }
    dfs();
    memset(dp,-0x3f,sizeof(dp));
    dp[][]=;
    for (int i=;i<=tim;i++)
    {
        int x=id[i];
        for (int j=;j<=p;j++) if (dp[i][j]>=)
        {
            if (j+w[x]<=p) dp[i+][j+w[x]]=max(dp[i+][j+w[x]],dp[i][j]+v[x]);
            dp[ed[x]+][j]=max(dp[ed[x]+][j],dp[i][j]);
            ans=max(ans,dp[i][j]);
        }
    }
    for (int i=;i<=p;i++) ans=max(ans,dp[tim+][i]);
    printf("%d\n",ans);
    return ;
}