BZOJ 1103 大都市MEG

求出DFS序
修路相当于区间减，点询
树状数组维护之

操作数要加(N-1)

#include <cstdio>
#include <cassert>
#include <string>

using namespace std;

int read(){
	int x=0, f=1;char ch=getchar();
	while(ch<'0' || ch>'9'){if(ch=='-')f=-f;ch=getchar();}
	while(ch>='0' && ch<='9'){x=x*10+(ch-'0');ch=getchar();}
	return x*f;
}

const int MAXN=250111;
const int MAXM=250111;

int N, M;

struct Vert{
	int FE;
	int Dps, Dpr;
	int Dep;
} V[MAXN];

struct Edge{
	int x, y, next;
} E[MAXN<<1];

int Ecnt=0;

void addE(int a, int b){
	++Ecnt;
	E[Ecnt].x=a;E[Ecnt].y=b;E[Ecnt].next=V[a].FE;V[a].FE=Ecnt;
}

int Dfn[MAXN], DFN=0;

void DFS(int at, int f=0){
	++DFN;Dfn[DFN]=at;
	V[at].Dps=DFN;
	
	for(int k=V[at].FE, to;k>0;k=E[k].next){
		to=E[k].y;
		if(to==f)	continue;
		V[to].Dep=V[at].Dep+1;
		DFS(to, at);
	}
	
	V[at].Dpr=DFN;
}

int C[MAXN];

int lowbit(int a){
	return a&(-a);
}

void Add(int at, int v){
	//cout << "Add " << at << " " << v << endl;
	for(int i=at;i<=N;i+=lowbit(i))
		C[i]+=v;
}

int Ask(int at){
	//cout << "Ask " << at << endl;
	int ret=0;
	for(int i=at;i>0;i-=lowbit(i))
		ret+=C[i];
	return ret;
}

char com[10];
int p, q;

int main(){
	
	N=read();
	for(int i=1, a, b;i<N;++i){
		a=read();b=read();
		addE(a, b);addE(b, a);
	}
	
	V[1].Dep=1;
	DFS(1);
	
	assert(DFN==N);
	
	for(int i=1, j;i<=N;++i){
		j=Dfn[i];
		Add(i, V[j].Dep);
		Add(i+1, -V[j].Dep);
	}
	
	M=read();M+=(N-1);/**/
	while(M--){
		scanf("%s", com);
		if(com[0]=='A'){
			p=read();q=read();
			if(V[p].Dep<V[q].Dep)	swap(p, q);
			Add(V[p].Dps, -1);
			Add(V[p].Dpr+1, 1);
		}
		else{
			p=read();/*p=Dfn[p];*/p=V[p].Dps;
			printf("%d
", Ask(p)-1/**/);
		}
	}
	
	return 0;
}