文章目录
所谓树上莫队,就是在树上的莫队
(逃)
传送门
解析
似乎就是树上的这道题
考虑如何转化为序列问题呢?
考虑dfs序
但是又一个问题。。。
似乎这条链的dfs序不连续啊树剖一下就好啦
考虑更阳间的方法
求出这棵树的欧拉序,在这个欧拉序上询问
那么我们发现,这样的话,其实会多算的部分就都会多算2遍
比如样例:
以1为根,欧拉序为:
13442231
那么我们考虑(4-3)
对应的序列就是:
44223
不在路径上的2恰好算了2次
所以我们可以利用异或的性质
还有一些特判的问题:
- lca不在端点时,需要额外计算lca
- 左端点不时lca时,需要额外计算左端点
画画图就很清楚了
代码
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=3e5+100;
const int M=1050;
const int mod=998244353;
const double eps=1e-6;
ll read(){
ll x=0,f=1;char c=getchar();
while(!isdigit(c)){if(c=='-') f=-1;c=getchar();}
while(isdigit(c)){x=x*10+c-'0';c=getchar();}
return x*f;
}
int n,m;
struct node{
int to,nxt;
}p[N<<1];
int fi[N],cnt;
void addline(int x,int y){
p[++cnt]=(node){y,fi[x]};fi[x]=cnt;
}
int pl[N][22],dep[N],siz[N];
ll v[N],w[N];
int pos[N],dfn[N],ed[N],Tim;
void dfs(int x,int f){
pl[x][0]=f;pos[x]=++Tim;dfn[Tim]=x;siz[x]=1;
for(int k=1;k<=18;k++) pl[x][k]=pl[pl[x][k-1]][k-1];
for(int i=fi[x];~i;i=p[i].nxt){
int to=p[i].to;
if(to==f) continue;
dep[to]=dep[x]+1;
dfs(to,x);
siz[x]+=siz[to];
}
dfn[++Tim]=x;ed[x]=Tim;
}
bool vis[N];
int a[N],tim,tot,from[N],to[N],id[N],bel[N];
int Lca(int x,int y){
if(dep[x]<dep[y]) swap(x,y);
for(int k=18;k>=0;k--){
if(!pl[x][k]||dep[pl[x][k]]<dep[y]) continue;
x=pl[x][k];
}
// printf("LCA:x=%d y=%d ",x,y);
if(x==y){
// printf("ares=%d\n",x);
return x;
}
// printf(" mid:x=%d y=%d ",x,y);
for(int k=18;k>=0;k--){
if(pl[x][k]==pl[y][k]) continue;
x=pl[x][k];y=pl[y][k];
}
// printf("x=%d res=%d\n",x,pl[x][0]);
return pl[x][0];
}
struct query{
int l,r,t,id,lca;
bool operator < (const query o)const{
if(bel[l]!=bel[o.l]) return bel[l]<bel[o.l];
else if(bel[r]!=bel[o.r]) return bel[r]<bel[o.r];
else return t<o.t;
}
}ask[N];
int l,r,t,bac[N];
ll now;
ll ans[N];
inline void work(int x){
x=dfn[x];
(vis[x]^=1)?now+=v[a[x]]*w[++bac[a[x]]]:now-=v[a[x]]*w[bac[a[x]]--];
}
int q;
int ww;
int main(){
memset(fi,-1,sizeof(fi));cnt=-1;
n=read();m=read();q=read();
for(int i=1;i<=m;i++) v[i]=read();
for(int i=1;i<=n;i++) w[i]=read();
for(int i=1;i<n;i++){
int x=read(),y=read();
addline(x,y);addline(y,x);
}
for(int i=1;i<=n;i++) a[i]=read();
dep[1]=1;dfs(1,0);
// for(int i=1;i<=Tim;i++) printf("%d ",dfn[i]);
// printf("\n");
ww=floor(pow(Tim,2.0/3));
for(int i=1;i<=Tim;i++){
bel[i]=(i-1)/ww+1;
}
for(int i=1;i<=q;i++){
int op=read(),x=read(),y=read();
if(op==0){
++tim;int o=x;
from[tim]=a[o];id[tim]=o;
a[o]=y;to[tim]=a[o];
}
else{
if(pos[x]>pos[y]) swap(x,y);
ask[++tot]=(query){pos[x],pos[y],tim,tot,pos[Lca(x,y)]};
}
// printf("i=%d op=%d x=%d y=%d\n",i,op,x,y);
}
sort(ask+1,ask+1+tot);
l=1;t=tim;
for(int i=1;i<=tot;i++){
int nl=ask[i].l,nr=ask[i].r,nt=ask[i].t,lca=ask[i].lca;
while(l<nl) work(l++);
while(l>nl) work(--l);
while(r<nr) work(++r);
while(r>nr) work(r--);
while(t<nt){
t++;
int o=id[t],f=0;
if((pos[o]<l&&l<=ed[o]&&ed[o]<=r)) f=1;
else if(l<=pos[o]&&pos[o]<=r&&r<ed[o]) f=2;
if(f==1) work(ed[o]);
else if(f==2) work(pos[o]);
a[o]=to[t];
if(f==1) work(ed[o]);
else if(f==2) work(pos[o]);
}
while(t>nt){
int o=id[t],f=0;
if((pos[o]<l&&l<=ed[o]&&ed[o]<=r)) f=1;
else if(l<=pos[o]&&pos[o]<=r&&r<ed[o]) f=2;
if(f==1) work(ed[o]);
else if(f==2) work(pos[o]);
a[o]=from[t];
if(f==1) work(ed[o]);
else if(f==2) work(pos[o]);
t--;
}
if(nl!=lca){
work(nl);
if(nr!=lca) work(lca);
}
// printf("id=%d (%d %d %d %d) res=%lld\n",ask[i].id,nl,nr,nt,lca,now);
ans[ask[i].id]=now;
if(nl!=lca){
work(nl);
if(nr!=lca) work(lca);
}
}
for(int i=1;i<=tot;i++) printf("%lld\n",ans[i]);
return 0;
}
/*
*/