jzoj5661 [GDOI2018Day1模拟4.17]药香沁鼻

Description

---

给定一棵树，每个节点有价值和体积。已知选择一个节点需要满足它的父亲节点也被选择，根据给出的背包容量求能获得的最大价值

Solution

---

题目是图片懒得放惹

这种东西似乎是一类题目，也就是树形依赖背包
我们把树转化成dfs序，设f[i,j]表示从后往前做到第i个dfs序，体积为j的最大价值

dp柿子：f[i,j]=max(f[i+1,j-w[x]]+v[i],f[i+size[x],j])
可以理解为当前节点选择与否决定了以它为根的子树是否能被选

一开始漏掉了不选的情况然后GG了（这都能漏

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)

const int N=5005;
const int E=20005;

struct edge {int x,y,next;} e[E];

int w[N],v[N],f[N][N*2];
int pos[N],tot;
int size[N];
int ls[N],edCnt;

void add_edge(int x,int y) {
    if (x!=y) {
        e[++edCnt]=(edge) {x,y,ls[x]}; ls[x]=edCnt;
    }
}

void dfs(int now) {
    pos[++tot]=now;
    size[now]=1;
    for (int i=ls[now];i;i=e[i].next) {
        dfs(e[i].y);
        size[now]+=size[e[i].y];
    }
}

int main(void) {
    freopen("medicine.in","r",stdin);
    freopen("medicine.out","w",stdout);
    int n,m; scanf("%d%d",&n,&m);
    rep(i,1,n) {
        int fa; scanf("%d%d%d",&w[i],&fa,&v[i]);
        add_edge(fa,i);
    }
    dfs(1);
    drp(ti,n,1) {
        int i=pos[ti];
        rep(j,0,m) {
            if (j<w[i]) f[ti][j]=std:: max(f[ti+size[i]][j],0);
            else {
                f[ti][j]=std:: max(f[ti][j],f[ti+1][j-w[i]]+v[i]);
                f[ti][j]=std:: max(f[ti][j],f[ti+size[i]][j]);
            }
        }
    }
    printf("%d\n", f[1][m]);
    return 0;
}