树状数组【bzoj1103】: [POI2007]大都市meg

1103: [POI2007]大都市meg

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

树状数组按照DFN序建树。

维护从当前点到根有多少条路被修过。

单步容斥记录一下dep就可以了。

code：

#include <iostream>
#include <cstdio>

using namespace std;

const int wx=500017;

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

char opt[5];
int n,m,num,tot;
int head[wx],sum[wx],dfn[wx],dep[wx],flag[wx];
int size[wx];

struct e{
	int nxt,to;
}edge[wx*2];

void F(int from,int to){
	edge[++num].nxt=head[from];
	edge[num].to=to;
	head[from]=num;
}

void add(int pos,int k){
	for(int i=pos;i<=n;i+=(i&-i)){
		sum[i]+=k;
	}
}

int query(int pos){
	int re=0;
	for(int i=pos;i>=1;i-=(i&-i)){
		re+=sum[i];
	}
	return re;
}

void dfs(int u,int f){
	dfn[u]=++tot;size[u]=1;
	for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].to;
		if(v==f)continue;
		dep[v]=dep[u]+1; 
		dfs(v,u);
		size[u]+=size[v];
	}
}

int main(){
	n=read();
	for(int i=1;i<n;i++){
		int x,y; 
		x=read(); y=read();
		F(x,y); F(y,x);
	}
	dfs(1,0); m=read();
	for(int i=1;i<=m+n-1;i++){
		scanf("%s",opt+1);
		if(opt[1]=='W'){
			int x;x=read();
			printf("%d\n",dep[x]-query(dfn[x]));
		}
		else{
			int x,y;
			x=read(); y=read();
			int zmj=(dep[x]>dep[y]?x:y);
			add(dfn[zmj],1);add(dfn[zmj]+size[zmj],-1);
		}
	}
}