BZOJ1103 [POI2007]大都市meg 【树剖】

1103: [POI2007]大都市meg

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 3038  Solved: 1593
[Submit][Status][Discuss]

Description

　　在经济全球化浪潮的影响下,习惯于漫步在清晨的乡间小路的邮递员Blue Mary也开始骑着摩托车传递邮件了。
不过，她经常回忆起以前在乡间漫步的情景。昔日，乡下有依次编号为1..n的n个小村庄，某些村庄之间有一些双
向的土路。从每个村庄都恰好有一条路径到达村庄1（即比特堡）。并且，对于每个村庄，它到比特堡的路径恰好
只经过编号比它的编号小的村庄。另外，对于所有道路而言，它们都不在除村庄以外的其他地点相遇。在这个未开
化的地方，从来没有过高架桥和地下铁道。随着时间的推移，越来越多的土路被改造成了公路。至今，Blue Mary
还清晰地记得最后一条土路被改造为公路的情景。现在，这里已经没有土路了——所有的路都成为了公路，而昔日
的村庄已经变成了一个大都市。 Blue Mary想起了在改造期间她送信的经历。她从比特堡出发，需要去某个村庄，
并且在两次送信经历的间隔期间,有某些土路被改造成了公路.现在Blue Mary需要你的帮助：计算出每次送信她需
要走过的土路数目。（对于公路，她可以骑摩托车；而对于土路，她就只好推车了。）

Input

　　第一行是一个数n(1 < = n < = 2 50000).以下n-1行，每行两个整数a，b（1 < =  a以下一行包含一个整数m
（1 < = m < = 2 50000），表示Blue Mary曾经在改造期间送过m次信。以下n+m-1行，每行有两种格式的若干信息
，表示按时间先后发生过的n+m-1次事件:若这行为 A a b(a若这行为 W a, 则表示Blue Mary曾经从比特堡送信到
村庄a。

Output

　　有m行，每行包含一个整数，表示对应的某次送信时经过的土路数目。

Sample Input

5
1 2
1 3
1 4
4 5
4
W 5
A 1 4
W 5
A 4 5
W 5
W 2
A 1 2
A 1 3

Sample Output

2
1
0
1

HINT

裸的树剖

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long int
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define Redge(u) for (int k = head[u]; k != -1; k = edge[k].next)
using namespace std;
const int maxn = 250005,maxm = 500005,INF = 1000000000;
inline int RD(){
	int out = 0,flag = 1; char c = getchar();
	while (c < 48 || c > 57) {if (c == '-') flag = -1; c = getchar();}
	while (c >= 48 && c <= 57) {out = (out << 1) + (out << 3) + c - '0'; c = getchar();}
	return out * flag;
}
int N,M,ls[maxn],rb[maxn],L,R,sum[4 * maxn];
int top[maxn],son[maxn],fa[maxn],dep[maxn],id[maxn],siz[maxn],cnt = 0;
void dfs1(int u,int f,int d){
	fa[u] = f; dep[u] = ++d; siz[u] = 1;
	for (int k = ls[u]; k; k = rb[k]){
		dfs1(k,u,d);
		siz[u] += siz[k];
		if (!son[u] || siz[k] > siz[son[u]]) son[u] = k;
	}
}
void dfs2(int u,bool flag){
	id[u] = ++cnt; top[u] = flag ? top[fa[u]] : u;
	if (son[u]) dfs2(son[u],true);
	for (int k = ls[u]; k; k = rb[k])
		if (k != son[u]) dfs2(k,false);
}
void add(int u,int l,int r){
	if (l == r) sum[u]++;
	else {
		int mid = l + r >> 1;
		if (mid >= L) add(u << 1,l,mid);
		else add(u << 1 | 1,mid + 1,r);
		sum[u] = sum[u << 1] + sum[u << 1 | 1];
	}
}
int Query(int u,int l,int r){
	if (l >= L && r <= R) return sum[u];
	else {
		int mid = l + r >> 1;
		if (mid >= R) return Query(u << 1,l,mid);
		else if (mid < L) return Query(u << 1 | 1,mid + 1,r);
		else return Query(u << 1,l,mid) + Query(u << 1 | 1,mid + 1,r);
	}
}
void cal(int u){
	int ans = dep[u] - 1;
	while (u){
		L = id[top[u]]; R = id[u];
		ans -= Query(1,1,N);
		u = fa[top[u]];
	}
	printf("%d\n",ans);
}
void solve(){
	M = RD() + N - 1; char cmd; int a,b;
	while (M--){
		cmd = getchar(); while (cmd != 'W' && cmd != 'A') cmd = getchar();
		if (cmd == 'W'){
			cal(RD());
		}else {
			a = RD(); b = RD(); if (a < b) swap(a,b);
			L = id[a]; add(1,1,N);
		}
	}
}
int main(){
	N = RD(); int a,b;
	REP(i,N - 1){
		a = RD(); b = RD(); if (a < b) swap(a,b);
		rb[a] = ls[b]; ls[b] = a;
	}
	dfs1(1,0,0);
	dfs2(1,0);
	solve();
	return 0;
}