[BZOJ4182]Shopping-树形依赖背包-点分治

Shopping

Description

马上就是小苗的生日了，为了给小苗准备礼物，小葱兴冲冲地来到了商店街。商店街有n个商店，并且它们之间的道路构成了一颗树的形状。

第i个商店只卖第i种物品，早苗对于这种物品的喜爱度是wi，物品的价格为ci，物品的库存是di。但是商店街有一项奇怪的规定：如果在商店u,v买了东西，并且有一个商店w在u到v的路径上，那么必须要在商店w买东西。小葱身上有m元钱，他想要尽量让小苗开心，所以他希望最大化小苗对买到物品的喜爱度之和。这种小问题对于小葱来说当然不在话下，但是他的身边没有电脑，于是他打电话给同为OI选手的你，你能帮帮他吗？

Input

输入第一行一个正整数T，表示测试数据组数。

对于每组数据，
第一行两个正整数n;m；
第二行n个非负整数w1,w2…wn；
第三行n个正整数c1,c2…cn；
第四行n个正整数d1,d2…dn；
接下来n-1行每行两个正整数u;v表示u和v之间有一条道路

Output

输出共T 行，每行一个整数，表示最大的喜爱度之和。

Sample Input

1
3 2
1 2 3
1 1 1
1 2 1
1 2
1 3

Sample Output

4

HINT

N<=500,M<=4000,T<=5,Wi<=4000,Di<=100

---

咱居然现在才学多重背包的单调队列优化……
已经弱得没救了……

---

思路:
看起来像经典的树形依赖背包。
对于树形依赖背包，有一个经典思路:
在树的dfs序上进行dp，若不选某个节点，则直接转移到子树中最大dfs序编号+1处，代表跳过整棵子树，否则向当前点dfs序+1处转移。

然而这题是无根的，不存在什么子树。

那么考虑点分治。
每次对于一棵分治出的子树，以分治重心为根，按经典的树形依赖背包做法，跑一边多重背包。
多重背包很慢，因此考虑引入单调队列来优化多重背包转移。

然后这题就做完了!

#include<bits/stdc++.h>
using namespace std;
const int N=509;
const int M=4009;

int n,m,ans;
int seg[N],id[N],ed[N],dfn;
int w[N],c[N],d[N],siz[N<<1];
int to[N<<1],nxt[N<<1],beg[N],tot=1;
bool ban[N<<1];

inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0' || '9'<ch){if(ch=='-')f=-1;ch=getchar();}
    while('0'<=ch && ch<='9')x=x*10+(ch^48),ch=getchar();
    return x*f;
}

inline void chkmax(int &a,int b){if(a<b)a=b;}

inline void add(int u,int v)
{
    to[++tot]=v;
    nxt[tot]=beg[u];
    beg[u]=tot;
}

inline int pre(int u,int fa,int sz,int &rt)
{
    int mx=0,szs=1;
    for(int i=beg[u];i;i=nxt[i])
        if(!ban[i] && to[i]!=fa)
        {
            chkmax(mx,(siz[i]=pre(to[i],u,sz,rt)));
            szs+=siz[i];siz[i^1]=n-siz[i];
        }
    chkmax(mx,sz-szs);
    if((mx<<1)<=sz)rt=u;
    return szs;
}

inline int dfs(int u,int fa)
{
    seg[id[u]=++dfn]=u;
    for(int i=beg[u];i;i=nxt[i])
        if(!ban[i] && to[i]!=fa)
            dfs(to[i],u);
    ed[u]=dfn;
}

inline void calc(int rt)
{
    dfn=0;dfs(rt,0);
    static int f[N][M],q[M],l,r;

    for(int i=1;i<=ed[rt]+1;i++)
        for(int j=0;j<=m;j++)
            f[i][j]=-1e9+7;

    f[1][0]=0;
    for(int i=1,u=seg[i];i<=ed[rt];i++,u=seg[i])
    {
        for(int x=0;x<c[u];x++)
            for(int j=x,l=0,r=0;j<=m;j+=c[u])
            {
                while(l<r && (j-q[l])>d[u]*c[u])++l;
                if(l<r)chkmax(f[i+1][j],f[i][q[l]]+(j-q[l])/c[u]*w[u]);
                while(l<r && f[i][j]>=f[i][q[r-1]]+(j-q[r-1])/c[u]*w[u])--r;
                q[r++]=j;
            }
        for(int j=0;j<=m;j++)
            chkmax(f[ed[u]+1][j],f[i][j]);
    }

    for(int i=0;i<=m;i++)
        chkmax(ans,f[ed[rt]+1][i]);
}

inline void work(int u,int sz)
{
    int rt=u;
    pre(u,0,sz,rt);
    calc(rt);
    for(int i=beg[rt];i;i=nxt[i])
        if(!ban[i])
        {
            ban[i]=ban[i^1]=1;
            work(to[i],siz[i]);
        }
}

int mina()
{
    ans=-1e9+7;tot=1;
    memset(beg,0,sizeof(beg));
    memset(ban,0,sizeof(ban));

    n=read();m=read();
    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=2,u,v;i<=n;i++)
    {
        u=read();v=read();
        add(u,v);add(v,u);
    }

    work(1,n);
    printf("%d\n",ans);
}
int main()
{
    int T=read();
    while(T--)
        mina();

    return 0;
}