钦妹的苹果树(appletree)
这是一道好题,要慎重地写啊。首先题目有点长概括一下。
题意:有k个小朋友,每个小朋友都有一个麻袋,他们麻袋的大小分别是1,2,...,k-1.这棵苹果树上有n个节点,每个节点上有无数个苹果,第i个节点上的每个苹果重量均为ti,甜度均为si。现在有Q组询问,每组询问描述一天的漫步,她们将从u走到v(因为是一棵树,所以路径当然唯一啦),沿途每个小朋友都会顺路采摘苹果,但重量和不能超过小朋友麻袋的大小。求所有小朋友能够获得的最大甜度和以及异或和。
看完题目第一感觉就是完全背包啊,枚举每一个小朋友,然后做一次完全背包,时间是O(Qk^2n),能得45分,可是考场上有些人神奇地拿了60分,真的厉害%。这题之所以叫做好题,是因为这个优化的想法我第一次见,很有帮助。由于在树上这个关键信息,这道题其实可以用倍增做,因为在一条路径上的最优取法是固定的,我们应该是可以求的。dp[i][j][k]表示从第i个节点向上跳2^j步的路径上重量为j的最大甜度,f[i][j]表示从第i个节点向上跳2^j步跳到的点。maxk[i]表示重量为k的最大甜度值。先构造出这棵苹果树,这个时候可以知道所有的f[i][j]。然后对于每一组询问u,v。我们能在求u和v的LCA的过程中处理出dp[i][j][k]和maxk[i],最后再做一次完全背包即可。
讲得不是很清楚,听不懂可以评论qwq...或者看代码
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#define N 40000+400
#define K 65
#define ll long long
using namespace std;
int cnt,n,k,Q,vet[N+N],head[N],next[N+N],t[N],s[N],ans[K],maxk[K],f[N][20],dp[N][20][K],deep[N];
ll ans1,ans2;
ll read(){
int f=1;ll num=0;char ch=getchar();
while(ch!='-' && ch<='0' && ch>='9') ch=getchar();
if(ch=='-') f=-1,ch=getchar();
while(ch>='0' && ch<='9') num=num*10+ch-'0',ch=getchar();
return f*num;
}
void add_edge(int u,int v){
vet[++cnt]=v;next[cnt]=head[u];head[u]=cnt;
}
void dfs(int u,int fa){
f[u][0]=fa;
dp[u][0][t[u]]=max(dp[u][0][t[u]],s[u]);
dp[u][0][t[fa]]=max(dp[u][0][t[fa]],s[fa]);
for(int i=1;i<=16;i++){
f[u][i]=f[f[u][i-1]][i-1];
for(int j=1;j<=k;j++) dp[u][i][j]=max(dp[u][i-1][j],dp[f[u][i-1]][i-1][j]);
}
for(int i=head[u];i;i=next[i]){
int v=vet[i];
if(v==fa) continue;
deep[v]=deep[u]+1;
dfs(v,u);
}
}
void LCA(int x,int y){
if(deep[x]<deep[y]) swap(x,y);
for(int i=16;i>=0;i--)
if(deep[f[x][i]]>=deep[y]){
for(int j=1;j<=k;j++) maxk[j]=max(maxk[j],dp[x][i][j]);
x=f[x][i];
}
if(x==y) return;
for(int i=16;i>=0;i--)
if(f[x][i]!=f[y][i]){
for(int j=1;j<=k;j++) maxk[j]=max(maxk[j],dp[x][i][j]);
for(int j=1;j<=k;j++) maxk[j]=max(maxk[j],dp[y][i][j]);
x=f[x][i],y=f[y][i];
}
for(int j=1;j<=k;j++) maxk[j]=max(maxk[j],dp[x][0][j]);
for(int j=1;j<=k;j++) maxk[j]=max(maxk[j],dp[y][0][j]);
}
int main()
{
freopen("appletree.in","r",stdin);
freopen("appletree.out","w",stdout);
n=read(),k=read();
for(int i=1;i<n;i++){
int u=read(),v=read();
add_edge(u,v);add_edge(v,u);
}
for(int i=1;i<=n;i++) t[i]=read();
for(int i=1;i<=n;i++) s[i]=read();
dfs(1,0);
Q=read();
while(Q--){
int u=read(),v=read();
for(int i=1;i<=k;i++) maxk[i]=ans[i]=0;
ans1=ans2=0;
maxk[t[u]]=s[u];maxk[t[v]]=s[v];
LCA(u,v);
for(int i=1;i<=k;i++)
for(int j=i;j<=k;j++) ans[j]=max(ans[j],ans[j-i]+maxk[i]);
for(int i=1;i<=k;i++) ans1=ans1+ans[i];
for(int i=1;i<=k;i++) ans2=ans2^ans[i];
printf("%lld %lld\n",ans1,ans2);
}
return 0;
}