【BZOJ】1103: [POI2007]大都市meg

#include <bits/stdc++.h>
using namespace std;

const int N=250005;
int n, ihead[N], cnt;
struct dat { int next, to; }e[N<<1];
void add(int u, int v) {
	e[++cnt].next=ihead[u]; ihead[u]=cnt; e[cnt].to=v;
	e[++cnt].next=ihead[v]; ihead[v]=cnt; e[cnt].to=u;
}
int dep[N], tot, FF[N], LL[N];
void dfs(int x, int fa) {
	FF[x]=++tot;
	for(int i=ihead[x]; i; i=e[i].next) if(e[i].to!=fa) dep[e[i].to]=dep[x]+1, dfs(e[i].to, x);
	LL[x]=tot;
}
int c[N];
void upd(int x, int s) { for(; x<=n; x+=x&-x) c[x]+=s; }
int sum(int x) { int ret=0; for(; x; x-=x&-x) ret+=c[x]; return ret; }
int main() {
	scanf("%d", &n);
	for(int i=0; i<n-1; ++i) { int u, v; scanf("%d%d", &u, &v); add(u, v); }
	dfs(1, 0);
	int m; scanf("%d", &m);
	for(int k=0; k<m+n-1; ++k) {
		int u, v;
		char s[2];
		scanf("%s%d", s, &u);
		if(s[0]==‘W‘) {
			printf("%d\n", sum(FF[u])+dep[u]); 
		}
		else {
			scanf("%d", &v);
			if(dep[u]<dep[v]) swap(u, v);
			upd(FF[u], -1);
			upd(LL[u]+1, 1);
		}
	}
	return 0;
}