JZOJ 5344. 摘果子

Description

Input

Output

Sample Input

7 9
39 6
13 2
22 6
7 4
-19 5
28 6
-17 1
2 1
3 2
4 1
5 4
6 2
7 3

Sample Output

52

Data Constraint

做法：一道非常经典的依赖背包问题，将树的dfs序求出后,设f[i][j]表示处理完i到n的数后毒素为j的最大价值。

状态转移如下：f[i][j]=max(f[i][j],f[i+1][j-p[i]]+v[i]) {选取当前果实}

　　　　　　　f[i][j]=max(f[i][j],f[i+size[x]])｛不选｝

size[x]表示dfs序中第i位的点的子树大小。

#include <cstdio>
#include <cstring>
#include <iostream>
#define N 2007
#define M 10007
using namespace std;
int f[N][M],ls[N],tot,v[N],p[N],n,m,size[N],d[N],cnt;
struct edge{
    int to,next;
}e[N*2];

void Add(int x,int y){
    e[++tot].to=y;
    e[tot].next=ls[x];
    ls[x]=tot;
}

void Dfs(int x,int fa){
    d[++cnt]=x;
    for (int i=ls[x];i;i=e[i].next){
        if (e[i].to==fa) continue;
        Dfs(e[i].to,x);
        size[x]+=size[e[i].to];
    }
    size[x]++;
}

int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) scanf("%d%d",&v[i],&p[i]);
    for (int i=1;i<=n-1;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        Add(x,y);
        Add(y,x);
    }
    Dfs(1,0);
    for (int i=n;i>=1;i--)
        for (int j=m;j>=0;j--){
            if (j-p[d[i]]>=0)    f[i][j]=max(f[i][j],f[i+1][j-p[d[i]]]+v[d[i]]);
            f[i][j]=max(f[i][j],f[i+size[d[i]]][j]); 
        }
    int ans=0;
    for (int i=0;i<=m;i++)
        ans=max(ans,f[1][i]);
    printf("%d",ans);
}