Description
S国有N个城市,编号从1到N。城市间用N-1条双向道路连接,满足
从一个城市出发可以到达其它所有城市。每个城市信仰不同的宗教,如飞天面条神教、隐形独角兽教、绝地教都是常见的信仰。为了方便,我们用不同的正整数代表各种宗教, S国的居民常常旅行。旅行时他们总会走最短路,并且为了避免麻烦,只在信仰和他们相同的城市留宿。当然旅程的终点也是信仰与他相同的城市。S国政府为每个城市标定了不同的旅行评级,旅行者们常会记下途中(包括起点和终点)留宿过的城市的评级总和或最大值。
在S国的历史上常会发生以下几种事件:
”CC x c”:城市x的居民全体改信了c教;
”CW x w”:城市x的评级调整为w;
”QS x y”:一位旅行者从城市x出发,到城市y,并记下了途中留宿过的城市的评级总和;
”QM x y”:一位旅行者从城市x出发,到城市y,并记下了途中留宿过
的城市的评级最大值。
由于年代久远,旅行者记下的数字已经遗失了,但记录开始之前每座城市的信仰与评级,还有事件记录本身是完好的。请根据这些信息,还原旅行者记下的数字。 为了方便,我们认为事件之间的间隔足够长,以致在任意一次旅行中,所有城市的评级和信仰保持不变。
Input
输入的第一行包含整数N,Q依次表示城市数和事件数。
接下来N行,第i+l行两个整数Wi,Ci依次表示记录开始之前,城市i的
评级和信仰。
接下来N-1行每行两个整数x,y表示一条双向道路。
接下来Q行,每行一个操作,格式如上所述。
Output
对每个QS和QM事件,输出一行,表示旅行者记下的数字。
Sample Input
5 6 3 1 2 3 1 2 3 3 5 1 1 2 1 3 3 4 3 5 QS 1 5 CC 3 1 QS 1 5 CW 3 3 QS 1 5 QM 2 4
Sample Output
8 9 11 3
Hint
N,Q < =10^5 , C < =10^5
数据保证对所有QS和QM事件,起点和终点城市的信仰相同;在任意时
刻,城市的评级总是不大于10^4的正整数,且宗教值不大于C。
树链剖分,同时根据不同的·宗教建立多棵线段树,分别统计答案。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define rep(i,j,k) for(int i=j;i<=k;i++)
#define per(i,j,k) for(int i=j;i>=k;i--)
#define loop(i,j,k) for (int i=j;i!=-1;i=k[i])
#define inone(x) scanf("%d",&x)
#define intwo(x,y) scanf("%d%d",&x,&y)
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
const int low(int x) { return x&-x; }
const int N = 2e5 + 10;
int n, m, x, y;
int a[N], b[N];
int ft[N], nt[N], u[N], sz;
int fa[N], mx[N], ct[N];
int dep[N], top[N], to[N];
int rt[N], L[N * 20], R[N * 20], o;
long long f[N * 20], g[N * 20];
char s[N];
void dfs(int x, int f)
{
ct[x] = 1; mx[x] = 0;
fa[x] = f; dep[x] = dep[f] + 1;
loop(i, ft[x], nt)
{
if (u[i] == f) continue;
dfs(u[i], x);
ct[x] += ct[u[i]];
if (ct[u[i]] > ct[mx[x]]) mx[x] = u[i];
}
}
void add(int &x, int l, int r, int u, int v)
{
if (!x) { x = ++o; L[o] = R[o] = f[o] = g[o] = 0; }
if (l == r) { f[x] = g[x] = v; return; }
int mid = l + r >> 1;
if (u <= mid) add(L[x], l, mid, u, v);
else add(R[x], mid + 1, r, u, v);
f[x] = f[L[x]] + f[R[x]];
g[x] = max(g[L[x]], g[R[x]]);
}
void Dfs(int x, int t)
{
top[x] = !t ? x : top[fa[x]];
to[x] = ++sz;
add(rt[b[x]], 1, n, to[x], a[x]);
if (mx[x]) Dfs(mx[x], 1);
loop(i, ft[x], nt)
{
if (u[i] == fa[x] || u[i] == mx[x]) continue;
Dfs(u[i], 0);
}
}
long long get(int x, int l, int r, int ll, int rr, int t)
{
if (ll <= l&&r <= rr) return t ? f[x] : g[x];
int mid = l + r >> 1;
long long res = 0, q;
if (ll <= mid)
{
q = get(L[x], l, mid, ll, rr, t);
if (t) res += q; else res = max(res, q);
}
if (rr > mid)
{
q = get(R[x], mid + 1, r, ll, rr, t);
if (t) res += q; else res = max(res, q);
}
return res;
}
long long get(int x, int y, int t)
{
long long res = 0, q;
int r = rt[b[x]];
for (; top[x] != top[y]; x = fa[top[x]])
{
if (dep[top[x]] < dep[top[y]]) swap(x, y);
q = get(r, 1, n, to[top[x]], to[x], t);
if (t) res += q; else res = max(res, q);
}
if (dep[x] > dep[y]) swap(x, y);
q = get(r, 1, n, to[x], to[y], t);
if (t) res += q; else res = max(res, q);
return res;
}
int main()
{
while (intwo(n, m) != EOF)
{
rep(i, 1, n) ft[i] = -1, intwo(a[i], b[i]);
ct[0] = dep[0] = sz = 0;
memset(rt, 0, sizeof(rt));
rep(i, 1, n - 1)
{
intwo(x, y);
u[sz] = y; nt[sz] = ft[x]; ft[x] = sz++;
u[sz] = x; nt[sz] = ft[y]; ft[y] = sz++;
}
dfs(1, 0); Dfs(1, o = sz = 0);
while (m--)
{
scanf("%s", s); intwo(x, y);
if (s[1] == 'C')
{
add(rt[b[x]], 1, n, to[x], 0);
add(rt[b[x] = y], 1, n, to[x], a[x]);
}
else if (s[1] == 'W')
{
add(rt[b[x]], 1, n, to[x], a[x] = y);
}
else
{
printf("%lld\n", get(x, y, s[1] == 'S'));
}
}
}
return 0;
}