bzoj 4182: Shopping

题意:

第i个商店只卖第i种物品,小苗对于这种物品的喜爱度是wi,物品的价格为ci,物品的库存是di。但是商店街有一项奇怪的规定:如果在商店u,v买了东西,并且有一个商店w在u到v的路径上,那么必须要在商店w买东西。小葱身上有m元钱,他想要尽量让小苗开心,所以他希望最大化小苗对买到物品的喜爱度之和。

题解:

暴力dp很好想,设 f[i][j] f [ i ] [ j ] 表示以节点i为根的子树且必选i用了j元的最大价值,然后每次把所有儿子合并上来,复杂度是 O(nm2) O ( n m 2 )
然后就不会了。
orz:beginend
code:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
const int inf=1<<28;
int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,w[510],c[510],d[510];
struct node{
    int y,next;
}a[1010];int len=0,last[510];
struct trnode{
    int son,tot;
}tr[510];
int f[510][4010],g[510][4010];
void ins(int x,int y)
{
    a[++len].y=y;
    a[len].next=last[x];last[x]=len;
}
void pre_node(int x,int fa)
{
    tr[x].tot=1;tr[x].son=0;
    for(int i=last[x];i;i=a[i].next)
    {
        int y=a[i].y;
        if(y==fa) continue;
        pre_node(y,x);tr[x].tot+=tr[y].tot;
        if(tr[tr[x].son].tot<tr[y].tot) tr[x].son=y;
    }
}
int q[4010][2],st,ed;
void ins(int *f,int x)
{
    for(int i=m;i>=c[x];i--) f[i]=f[i-c[x]]+w[x];
    for(int i=0;i<c[x];i++) f[i]=-inf;
    for(int i=0;i<c[x];i++)
    {
        st=1;ed=0;
        for(int j=i;j<=m;j+=c[x])
        {
            while(st<=ed&&q[ed][2]<f[j]-(j-i)/c[x]*w[x]) ed--;
            while(st<=ed&&q[st][1]<(j-i)/c[x]-d[x]+1) st++;
            q[++ed][1]=(j-i)/c[x];q[ed][2]=f[j]-q[ed][1]*w[x];
            f[j]=q[ed][1]*w[x]+q[st][2];
        }
    }
}
void solve(int x,int fa,int rt)
{
    ins(f[rt],x);
    for(int i=last[x];i;i=a[i].next)
    {
        if(a[i].y==fa) continue;
        for(int j=0;j<=m;j++) g[x][j]=f[rt][j];
        solve(a[i].y,x,rt);
        for(int j=0;j<=m;j++) f[rt][j]=max(f[rt][j],g[x][j]);
    }
}
void dfs(int x,int fa)
{
    for(int i=last[x];i;i=a[i].next)
        if(a[i].y!=fa) dfs(a[i].y,x);
    for(int i=0;i<=m;i++) f[x][i]=max(0,f[tr[x].son][i]);
    ins(f[x],x);
    for(int i=last[x];i;i=a[i].next)
    {
        int y=a[i].y;
        if(y==fa||y==tr[x].son) continue;
        for(int j=0;j<=m;j++) g[x][j]=f[x][j];
        solve(y,x,x);
        for(int j=0;j<=m;j++) f[x][j]=max(f[x][j],g[x][j]);
    }
}
int main()
{
    int T=read();
    while(T--)
    {
        n=read();m=read();
        len=0;memset(last,0,sizeof(last));
        for(int i=1;i<=n;i++) w[i]=read();
        for(int i=1;i<=n;i++) c[i]=read();
        for(int i=1;i<=n;i++) d[i]=read();
        for(int i=1;i<n;i++)
        {
            int x=read(),y=read();
            ins(x,y);ins(y,x);
        }
        memset(f,0,sizeof(f));
        pre_node(1,0);dfs(1,0);
        int ans=0;
        for(int i=1;i<=n;i++) ans=max(ans,f[i][m]);
        printf("%d\n",ans);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值