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;
}