【BZOJ】3052: [wc2013]糖果公园

#include <bits/stdc++.h>
using namespace std;
const int N=100005, M=100005;
typedef long long ll;
inline int getint() { int x=0; char c=getchar(); while(c<‘0‘||c>‘9‘) c=getchar(); while(c>=‘0‘&&c<=‘9‘) (x*=10)+=c-‘0‘, c=getchar(); return x; }
inline void print(ll a) { if(!a) return; print(a/10); putchar(‘0‘+(a%10));}
int ihead[N], cnt, blo[N<<1], f[N][17], FF[N], LL[N], tot, dep[N], cal[N], pos[N<<1], col[M], n, m, W[N], V[M], qu, n_ask, n_tm, col_pre[N];
ll Ans[M], ans;
bool st[N];
struct E { int next, to; }e[N<<1];
struct Q { int x, y, lca, id, tm; }q[M];
struct T { int x, y, last; }Time[M];
inline bool cmp(const Q &a, const Q &b) { return blo[a.x]==blo[b.x]?(blo[a.y]==blo[b.y]?a.tm<b.tm:blo[a.y]<blo[b.y]):blo[a.x]<blo[b.x]; }
inline void add(int x, int y) { e[++cnt]=(E){ihead[x], y}; ihead[x]=cnt; e[++cnt]=(E){ihead[y], x}; ihead[y]=cnt; }
void dfs(int x) {
	pos[FF[x]=++tot]=x;
	for(int i=1; i<=16; ++i) f[x][i]=f[f[x][i-1]][i-1];
	for(int i=ihead[x]; i; i=e[i].next) if(e[i].to!=f[x][0])
		dep[e[i].to]=dep[x]+1, f[e[i].to][0]=x, dfs(e[i].to);
	pos[LL[x]=++tot]=x;
}
inline int LCA(int x, int y) {
	if(dep[x]<dep[y]) swap(x, y);
	int d=dep[x]-dep[y];
	for(int i=16; i>=0; --i) if((d>>i)&1) x=f[x][i]; if(x==y) return x;
	for(int i=16; i>=0; --i) if(f[x][i]!=f[y][i]) x=f[x][i], y=f[y][i];
	return f[x][0];
}
inline void update(int x) {
	if(st[x]) { ans-=(ll)V[col[x]]*W[cal[col[x]]]; --cal[col[x]]; }
	else { ++cal[col[x]]; ans+=(ll)V[col[x]]*W[cal[col[x]]];  }
	st[x]=!st[x];
}
inline void change(int a, int b) {
	if(st[a]) { update(a); col[a]=b; update(a); }
	else col[a]=b;
}
inline void timechange(int &now, int goal) {
	while(now<goal) ++now, change(Time[now].x, Time[now].y);
	while(now>goal) change(Time[now].x, Time[now].last), --now;
}
void work() {
	int l=1, r=0, now=0, nl, nr;
	sort(q+1, q+1+n_ask, cmp);
	for(int i=1; i<=n_ask; ++i) {
		nl=q[i].x; nr=q[i].y;
		timechange(now, q[i].tm);
		while(l<nl) update(pos[l++]);
		while(l>nl) update(pos[--l]);
		while(r<nr) update(pos[++r]);
		while(r>nr) update(pos[r--]);
		if(q[i].lca) update(q[i].lca);
		Ans[q[i].id]=ans;
		if(q[i].lca) update(q[i].lca);
	}
	while(r>=l) update(pos[r--]);
}
void pre() {
	dfs((n+1)>>1);
	int nn=n<<1, sq=pow(nn, 2.0/3)*0.5;
	for(int i=1; i<=nn; ++i) blo[i]=(i-1)/sq;
	for(int i=1; i<=qu; ++i) {
		int type=getint(), x=getint(), y=getint();
		if(!type) {
			++n_tm;
			Time[n_tm]=(T){x, y, col_pre[x]}; col_pre[x]=y;
			continue;
		}
		++n_ask;
		if(FF[x]>FF[y]) swap(x, y);
		int lca=LCA(x, y);
		if(lca==x) 	q[n_ask]=(Q){FF[x], FF[y], 0, n_ask, n_tm};
		else 		q[n_ask]=(Q){LL[x], FF[y], lca, n_ask, n_tm};
	}
}
int main() {
	n=getint(); m=getint(); qu=getint();
	for(int i=1; i<=m; ++i) V[i]=getint();
	for(int i=1; i<=n; ++i) W[i]=getint();
	for(int i=1; i<n; ++i) {
		int x=getint(), y=getint();
		add(x, y);
	}
	for(int i=1; i<=n; ++i) col[i]=col_pre[i]=getint();
	pre();
	work();
	for(int i=1; i<=n_ask; ++i) print(Ans[i]), puts("");
	return 0;
}