bzoj 1103: [POI2007]大都市meg (dfs 序 + 树状数组）

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

给你一棵树。 除了一号点，每个点都有一个权值 1 。

每次询问，求一号店到 x 号点的路径上权值和。

或者修改某个点的权值为 0 。 

思路： 

树状数组 + dfs 序。 

因为修改每个点，影响的只会是这个点的孩子。 

我们想着怎么只修改孩子就可以了。 

那就是 dfs 序。 dfs 之后，孩子的序号都是连续的。 

就可以用树状数组维护了。

 

 

 

 

#include<bits/stdc++.h>
#define low(x) (x & (-x))
using namespace std;
const int N = 3e5;
int Head[N*2],Next[N*2],To[N*2],cnt;
int c[N],dep[N],l[N],r[N],tim;
int n,m;
void add(int u,int v){
    Next[++cnt] = Head[u];
    To[cnt] = v;
    Head[u] = cnt;
}
void dfs(int u,int fa){
    l[u] = ++tim; 
    for (int i = Head[u]; i; i = Next[i]){
        int v = To[i];
        if (v == fa) continue;
        dfs(v,u);
    }
    r[u] = tim;
}
void Add(int x, int w){for (; x <= n; x += low(x)) c[x] += w; }
int sum(int x){int sum = 0; for (; x; x -= low(x)) sum += c[x]; return sum;}
int main(){
    int x,y;
    scanf("%d",&n);
    for (int i = 1; i < n; ++i){
        scanf("%d%d",&x,&y);
        add(x,y),add(y,x);
    } 
    dfs(1,0);
    for (int i = 2; i <= n; ++i)
        Add(l[i],1),Add(r[i]+1,-1);
    char ch[5];
    scanf("%d",&m);
    for (int i = 1; i < n+m; ++i){
        scanf("%s",ch);
        if (ch[0] == 'W'){
            scanf("%d",&x);
            int ans = sum(l[x]);
            printf("%d\n",ans);
        }else{
            scanf("%d%d",&x,&y);
            if (l[x] < l[y]) x = y;
            Add(l[x],-1); Add(r[x]+1,1);
        }
    }
    return 0;
}