UOJ #58 【WC2013】 糖果公园

时间:2021-05-13 14:59:41

题目链接:糖果公园

  听说这是一道树上莫队的入门题,于是我就去写了……顺便复习了一下莫队的各种姿势。

  首先,我们要在树上使用莫队,那么就需要像序列一样给树分块。这个分块的过程就是王室联邦这道题(vfleaking大神的博客里也有讲),当然也可以按照\(dfs\)序什么的进行分块。

  但我还是想在这里讲一下我的分块做法(第一种)。我们可以先设一个块的大小$S$,然后对整棵树进行$dfs$。每次我们$dfs$完一棵子树回溯时,我们就把这个点$u$加入到一个队列中,更新$fa_u$的$siz$。这里$siz_u$记录的是以$u$为根的子树中为分好块的节点数。当我们发现$siz_u$超过了$S$时,我们就可以给这些节点新建一个块。最后还会剩下不超过$S$个节点,我们可以新建一个块,也可以把这些点丢到最后新建的那个块中。如果采用后一种方式,那么就保证了每个块的大小都在$[S,3S)$之间且最多只有一个块的大小会大于$2S$。于是我们就得到了一种优美的分块方法。(如果还不懂的话可以去看代码)

  然后,这道题带了修改(其实并没有什么不同,只是需要修改块的大小以及多一个时间流逝的操作),于是给每个操作多一个时间戳即可。树上莫队和序列莫队最大的不同就在于如何转移。也就是我们如何由路径$(u_i,v_j)$走到路径$(u_j,v_j)$上去。其实就是先把$u_i$走到$u_j$,再把$v_i$走到$v_j$即可。注意,这里为了方便考虑,上述路径均没有把深度最小的点给算进去。这里没有讲清楚……还是往后看吧……

  至于证明吗……我们先设几个东西(下列内容参考vfleaking大神的博客)……

  设$S(u,v)$表示路径$(u,v)$上所有点的集合,$T(u,v)$表示$S(u,v)$去掉$lca(u,v)$之后剩下的点集。那么显然有:$$T(u,v)=S(root,u) \  xor \  S(root,v)$$

  这里的$xor$指的是集合的对称差。简单来说就是把出现偶数次的节点给删掉。这个操作显然和异或差不多,性质也差不多。

  于是我们可以考虑一下从$T(u_i,v_i)$转移到$T(u_j,v_j)$需要改变一些什么。$$\because T(u_i,v_i)=S(root,u_i)\ xor \ S(root,v_i)\quad T(u_j,v_j)=S(root,u_j)\ xor \ S(root,v_j)$$ $$\therefore T(u_i,v_i)\ xor \ T(u_j,v_j)=T(u_i,u_j)\ xor \ T(v_i,v_j)$$ $$\therefore T(u_j,v_j)=T(u_i,v_i)\ xor \ T(u_i,u_j)\ xor \ T(v_i,v_j)$$

  于是我们从$T(u_i,v_i)$转移到$T(u_j,v_j)$只需要把点集$T(u_i,u_j)$和$T(v_i,v_j)$中的点存在性全部取反即可。

  至于如何$O(1)$地进行插入和删除就不需要我多说了吧……

  其实我感觉有vfleaking大神的博客应该不需要我的这一篇才对

  下面贴代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define File(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)
#define maxn 100010 using namespace std;
typedef long long llg; int n,m,q,V[maxn],co[maxn],num[maxn];
int fa[maxn][17],dep[maxn],siz[maxn];
int head[maxn],next[maxn<<1],to[maxn<<1],tt;
int qt[maxn],fr[maxn],qto,nblo,sizblo;
llg sw[maxn],now,ans[maxn]; bool vis[maxn];
struct data{
int u,v,t;
bool operator < (const data &h)const{
if(fr[u]!=fr[h.u]) return fr[u]<fr[h.u];
if(fr[v]!=fr[h.v]) return fr[v]<fr[h.v];
return t<h.t;
}
}s[maxn],cha[maxn<<1]; int getint(){
int w=0;bool q=0;
char c=getchar();
while((c>'9'||c<'0')&&c!='-') c=getchar();
if(c=='-') c=getchar(),q=1;
while(c>='0'&&c<='9') w=w*10+c-'0',c=getchar();
return q?-w:w;
} void link(int x,int y){
to[++tt]=y;next[tt]=head[x];head[x]=tt;
to[++tt]=x;next[tt]=head[y];head[y]=tt;
} void dfs(int u,int ff){
qt[++qto]=u;
fa[u][0]=ff; dep[u]=dep[ff]+1;
for(int i=1,now=2;now<dep[u];i++,now<<=1)
fa[u][i]=fa[fa[u][i-1]][i-1];
for(int i=head[u],v;v=to[i],i;i=next[i])
if(v!=ff){
dfs(v,u); siz[u]+=siz[v];
if(siz[u]>=sizblo){
nblo++; siz[u]=0;
while(qt[qto]!=u) fr[qt[qto--]]=nblo;
}
}
siz[u]++;
} void change(int u){
now-=V[co[u]]*sw[num[co[u]]];
if(vis[u]) num[co[u]]--,vis[u]=0;
else num[co[u]]++,vis[u]=1;
now+=V[co[u]]*sw[num[co[u]]];
} void timego(int x){
if(!cha[x].t) return;
bool ww=vis[cha[x].t];
if(ww) change(cha[x].t);
co[cha[x].t]=cha[x].v;
if(ww) change(cha[x].t);
} int lca(int u,int v){
if(dep[u]<dep[v]) swap(u,v); int t=0;
for(int now=1;now<dep[u];now<<=1) t++;t--;
for(int i=t;i>=0;i--)
if(dep[fa[u][i]]>=dep[v]) u=fa[u][i];
if(u==v) return u;
for(int i=t;i>=0;i--)
if(fa[u][i]!=fa[v][i])
u=fa[u][i],v=fa[v][i];
return fa[u][0];
} void query(int u,int v){
int g=lca(u,v);
while(u!=g) change(u),u=fa[u][0];
while(v!=g) change(v),v=fa[v][0];
} int main(){
File("a");
n=getint(); m=getint(); q=getint();
for(int i=1;i<=m;i++) V[i]=getint();
for(int i=1;i<=n;i++) sw[i]=sw[i-1]+getint();
for(int i=1;i<n;i++) link(getint(),getint());
sizblo=pow(n,2.0/3.0)+1; dfs(1,0); tt=0;
for(int i=1;i<=n;i++) if(!fr[i]) fr[i]=nblo;
for(int i=1;i<=n;i++) co[i]=num[i]=getint();
for(int i=1,ty;i<=q;i++){
ty=getint(); s[i].t=++tt;
s[i].u=getint(); s[i].v=getint();
if(!ty){
cha[i].u=num[s[i].u]; cha[i].t=s[i].u;
cha[i].v=num[s[i].u]=s[i].v;
cha[i+q]=cha[i]; swap(cha[i+q].u,cha[i+q].v);
}
else if(fr[s[i].u]>fr[s[i].v]) swap(s[i].u,s[i].v);
}
for(int i=1;i<=n;i++) num[i]=0;
sort(s+1,s+q+1);
for(int i=1,nt=0,u=1,v=1,g;i<=q;i++){
if(cha[s[i].t].t) continue;
g=lca(s[i].u,s[i].v);
while(nt<s[i].t) timego(++nt);
while(nt>s[i].t) timego(q+(nt--));
query(u,s[i].u); query(v,s[i].v);
change(g); ans[s[i].t]=now; change(g);
u=s[i].u; v=s[i].v;
}
for(int i=1;i<=q;i++)
if(!cha[i].t) printf("%lld\n",ans[i]);
return 0;
}