【bzoj3052】[wc2013]糖果公园 带修改树上莫队

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 100010
using namespace std;
const int b = 1200;
typedef long long ll;
int v[N] , w[N] , head[N] , to[N << 1] , next[N << 1] , ce , type[N];
int fa[N][20] , deep[N] , log[N] , tot , sta[N] , top , bl[N] , num;
int pc[N] , wc[N] , vc[N] , vis[N] , cnt[N];
ll ans , ret[N];
struct data
{
	int lp , rp , cp , id;
	bool operator<(const data &a)const
	{
		return bl[lp] == bl[a.lp] ? bl[rp] == bl[a.rp] ? cp < a.cp : bl[rp] < bl[a.rp] : bl[lp] < bl[a.lp];
	}
}a[N];
void add(int x , int y)
{
	to[++ce] = y , next[ce] = head[x] , head[x] = ce;
}
void dfs(int x)
{
	int i , now = top;
	for(i = 1 ; (1 << i) <= deep[x] ; i ++ ) fa[x][i] = fa[fa[x][i - 1]][i - 1];
	for(i = head[x] ; i ; i = next[i])
	{
		if(to[i] != fa[x][0])
		{
			fa[to[i]][0] = x , deep[to[i]] = deep[x] + 1 , dfs(to[i]);
			if(top - now >= b)
			{
				num ++ ;
				while(top != now) bl[sta[top -- ]] = num;
			}
		}
	}
	sta[++top] = x;
}
int lca(int x , int y)
{
	int i;
	if(deep[x] < deep[y]) swap(x , y);
	for(i = log[deep[x] - deep[y]] ; ~i ; i -- )
		if(deep[x] - deep[y] >= (1 << i))
			x = fa[x][i];
	if(x == y) return x;
	for(i = log[deep[x]] ; ~i ; i -- )
		if(deep[x] >= (1 << i) && fa[x][i] != fa[y][i])
			x = fa[x][i] , y = fa[y][i];
	return fa[x][0];
}
void rev(int x)
{
	if(vis[x]) ans -= (ll)w[cnt[type[x]]] * v[type[x]] , cnt[type[x]] -- , vis[x] = 0;
	else cnt[type[x]] ++ , ans += (ll)w[cnt[type[x]]] * v[type[x]] , vis[x] = 1;
}
int main()
{
	int n , m , q , i , j , x , y , opt , cc = 0 , ln = 1 , rn = 1 , now = 0;
	scanf("%d%d%d" , &n , &m , &q);
	for(i = 1 ; i <= m ; i ++ ) scanf("%d" , &v[i]);
	for(i = 1 ; i <= n ; i ++ ) scanf("%d" , &w[i]);
	for(i = 2 ; i <= n ; i ++ ) scanf("%d%d" , &x , &y) , add(x , y) , add(y , x) , log[i] = log[i >> 1] + 1;
	dfs(1);
	while(top) bl[sta[top -- ]] = num;
	for(i = 1 ; i <= n ; i ++ ) scanf("%d" , &type[i]);
	for(i = 1 ; i <= q ; i ++ )
	{
		scanf("%d" , &opt);
		if(opt)
		{
			scanf("%d%d" , &a[i - cc].lp , &a[i - cc].rp) , a[i - cc].cp = cc , a[i - cc].id = i - cc;
			if(bl[a[i - cc].lp] > bl[a[i - cc].rp]) swap(a[i - cc].lp , a[i - cc].rp);
		}
		else cc ++ , scanf("%d%d" , &pc[cc] , &vc[cc]) , wc[cc] = type[pc[cc]] , type[pc[cc]] = vc[cc];
	}
	for(i = cc ; i ; i -- ) type[pc[i]] = wc[i];
	sort(a + 1 , a + q - cc + 1);
	for(i = 1 ; i <= q - cc ; i ++ )
	{
		x = lca(ln , a[i].lp);
		for(j = ln ; j != x ; j = fa[j][0]) rev(j);
		for(j = a[i].lp ; j != x ; j = fa[j][0]) rev(j);
		x = lca(rn , a[i].rp);
		for(j = rn ; j != x ; j = fa[j][0]) rev(j);
		for(j = a[i].rp ; j != x ; j = fa[j][0]) rev(j);
		ln = a[i].lp , rn = a[i].rp , x = lca(ln , rn) , rev(x);
		while(now < a[i].cp)
		{
			now ++ ;
			if(vis[pc[now]]) cnt[vc[now]] ++ , ans += (ll)w[cnt[vc[now]]] * v[vc[now]] - (ll)w[cnt[wc[now]]] * v[wc[now]] , cnt[wc[now]] -- ;
			type[pc[now]] = vc[now];
		}
		while(now > a[i].cp)
		{
			if(vis[pc[now]]) cnt[wc[now]] ++ , ans += (ll)w[cnt[wc[now]]] * v[wc[now]] - (ll)w[cnt[vc[now]]] * v[vc[now]] , cnt[vc[now]] -- ;
			type[pc[now]] = wc[now];
			now -- ;
		}
		ret[a[i].id] = ans;
		rev(x);
	}
	for(i = 1 ; i <= q - cc ; i ++ ) printf("%lld\n" , ret[i]);
	return 0;
}