YBTOJ&洛谷P4074：糖果公园（树上莫队）

所谓树上莫队，就是在树上的莫队

（逃）
传送门

解析

似乎就是树上的这道题
考虑如何转化为序列问题呢?
考虑dfs序
但是又一个问题。。。
似乎这条链的dfs序不连续啊
树剖一下就好啦
考虑更阳间的方法
求出这棵树的欧拉序，在这个欧拉序上询问
那么我们发现，这样的话，其实会多算的部分就都会多算2遍
比如样例：

以1为根，欧拉序为：
13442231
那么我们考虑（4-3）
对应的序列就是：

44223

不在路径上的2恰好算了2次
所以我们可以利用异或的性质
还有一些特判的问题：

1. lca不在端点时，需要额外计算lca
2. 左端点不时lca时，需要额外计算左端点
   画画图就很清楚了

update:

上面那个特判是洛谷野生题解的，有点过于阴间了…
ybt的实现优美的多（至少易于记忆）
设一个点进入和离开dfs的时间分别为 $in(x),out(x)$。
令 $in(x)<in(y)$，分情况讨论：

1. $x$ 是 $y$ 的祖先，那么答案就是 $[in(x),in(y)]$。
2. $x$ 不是 $y$ 的祖先，那么在 $[out(x),in(y)]$ 的基础上，还要加上 $lca(x,y)$。

代码

#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;
}
/*

*/