JZOJ5344. 【NOIP2017模拟9.3A组】摘果子 树上依赖背包

92 篇文章 1 订阅

算法听起来高大上其实是个水题。
题意:给出一棵树,每个点有花费和收益,选择一个点必须选择他的父亲。从根出发,问代价在m以内的最高收益是多少.n,m<=2e3.

其实很明显的背包,就是依赖关系有点难搞(其实是我太弱)。
首先既然选择一个点必须选择他的父亲,那么我们所有的转移都要强制性,即:
f[x][j]=f[x][jw[j]]+v[j]
然后,如果要选择,那么整一个条链都要选择。则有:
f[x][j]=max(f[x][j],f[v][j])
答案就是 max(0,f[1][m])
注意是逆推,从下往上。
代码十分简洁= =

#include<cstdio>
#include<algorithm>
#include<cstring>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2e3+5;
typedef long long ll;
const ll inf=1e15;
ll f[N][N];
int head[N<<1],next[N<<1],go[N<<1];
int n,m,val[N],p[N],tot;
inline void add(int x,int y)
{
    go[++tot]=y;
    next[tot]=head[x];
    head[x]=tot;
}
inline void dfs(int x,int fa)
{
    fd(i,m,p[x])f[x][i]=f[x][i-p[x]]+val[x];
    fo(i,0,p[x]-1)f[x][i]=-inf;
    for(int i=head[x];i;i=next[i])
    {
        int v=go[i];
        if (v!=fa)
        {
            memcpy(f[v],f[x],sizeof(f[v]));
            dfs(v,x);
            fo(j,0,m)
            f[x][j]=max(f[x][j],f[v][j]);
        }
    }
    //if (!flag)memset(f[x],0,sizeof(f[x]));
}
int main()
{
    //freopen("T2.in","r",stdin);
    //freopen("T2.out","w",stdout);
    scanf("%d%d",&n,&m);
    fo(i,1,n)scanf("%d%d",&val[i],&p[i]);
    fo(i,1,n-1)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(1,0);
    printf("%lld\n",max(f[1][m],0ll));
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值