bzoj 1103: [POI2007]大都市meg 树状数组

       一句话题意：将边权修改为0（初始为1）和查询某个点到1的距离。

       一开始预处理没有问题吧，然后修改边权只对它管辖的子树有影响。相当于区间修改。然后查分一下用树状数组即可。

AC代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 250005
using namespace std;

int n,m,tot,dfsclk,c[N],bg[N],ed[N],fst[N],pnt[N],nxt[N];
int read(){
	int x=0; char ch=getchar();
	while (ch<'0' || ch>'9') ch=getchar();
	while (ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
	return x;
}
void add(int x,int y){
	pnt[++tot]=y; nxt[tot]=fst[x]; fst[x]=tot;
}
void ins(int x,int t){
	for (; x<=n; x+=x&-x) c[x]+=t;
}
int getsum(int x){
	int t=0; for (; x; x-=x&-x) t+=c[x]; return t;
}
void dfs(int x){
	bg[x]=++dfsclk; int p;
	for (p=fst[x]; p; p=nxt[p]) dfs(pnt[p]);
	ed[x]=dfsclk;
}
int main(){
	n=read(); int i,x,y;
	for (i=1; i<n; i++){
		x=read(); y=read(); add(x,y);
	}
	dfs(1); m=read()+n-1; char ch;
	for (i=2; i<=n; i++){ ins(bg[i],1); ins(ed[i]+1,-1); }
	while (m--){
		ch=getchar(); while (ch<'A' || ch>'Z') ch=getchar();
		if (ch=='W') printf("%d\n",getsum(bg[read()])); else{
			x=read(); y=read();
			ins(bg[y],-1); ins(ed[y]+1,1);
		}
	}
	return 0;
}

by lych

2016.3.15