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

题目描述

给出一棵n个点的树，每个点有一个点权，点权范围为1~m。支持两种操作：(1)修改一个点的点权 (2)对于一条路径，求$\sum\limits_{i=1}^m\sum\limits_{j=1}^{s_i}V_iW_j$，其中$s_i$表示这条链上权值为i的点数。

输入

输出

样例输入

4 3 5
1 9 2
7 6 5 1
2 3
3 1
3 4
1 2 3 2
1 1 2
1 4 2
0 2 1
1 1 2
1 4 2

样例输出

84
131
27
84

---

题解

带修改树上莫队

带修改树上莫队——普通莫队的 带修改进化版+上树树上进化版。

带修改莫队：每块大小$n^{\frac 23}$，分别以左端点所在块、右端点所在块、时间（这次询问之前的修改次数）为第一、二、三关键字排序，然后暴力移动三个指针。

树上莫队：分块方法改成树分块（树分块方法参考 王室联邦），自然序改成DFS序（然而带修改莫队用不到自然序，这里只是提一嘴），暴力移动指针。这里可以把点权加到边权上（链上LCA只在统计答案是算上），以避免特判。

把它们结合起来本题就做完了。把树分块，然后按照带修改莫队处理即可。

几点注意：

求LCA要用非朴素LCA（因为不仅仅在移动指针时需要求LCA，点权加到边权上时统计答案也需要处理LCA）

修改要记录是从什么改到什么，因为有反向修改（时间指针左移）操作。

答案会爆int，故需要long long。

时间复杂度$O(n^{\frac 53})$

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