NOIP2017模拟赛 大都市（dfs序+树状数组）

O大都市
时间限制 : - MS 空间限制 : 165536 KB
评测说明 : 2s
问题描述

在经济全球化浪潮的影响下,习惯于漫步在清晨的乡间小路的邮递员Blue Mary也开始骑着摩托车传递邮件了。
不过，她经常回忆起以前在乡间漫步的情景。昔日，乡下有依次编号为1..n的n个小村庄，某些村庄之间有一些双向的土路。从每个村庄都恰好有一条路径到达村庄1（即比特堡）。并且，对于每个村庄，它到比特堡的路径恰好只经过编号比它的编号小的村庄。

另外，对于所有道路而言，它们都不在除村庄以外的其他地点相遇。在这个未开化的地方，从来没有过高架桥和地下铁道。随着时间的推移，越来越多的土路被改造成了公路。至今，Blue Mary还清晰地记得最后一条土路被改造为公路的情景。现在，这里已经没有土路了——所有的路都成为了公路，而昔日的村庄已经变成了一个大都市。
Blue Mary想起了在改造期间她送信的经历。她从比特堡出发，需要去某个村庄，并且在两次送信经历的间隔期间,有某些土路被改造成了公路.
现在Blue Mary需要你的帮助：计算出每次送信她需要走过的土路数目。（对于公路，她可以骑摩托车；而对于土路，她就只好推车了。）

输入格式

　第一行是一个数n(1 < = n < = 2 50000).
以下n-1行，每行两个整数a，b（1 < =a

题解

dfs序初始化每个节点的in与ou值
对于每次修改更改in[x]-1的值与ou[x]值即可

类似与差分数组，用树状数组维护即可

代码

#include<stdio.h>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
#include<cstdio>
#include<iostream>
#include<vector>
#include<map>
using namespace std;
#define maxn 260005
int n,m;
int d[maxn];
int in[maxn],ou[maxn];
int low(int x) {return (x&(-x));}
void md(int x,int s){
    int i,j,k;
    for(i=x;i<=n;i+=low(i)){
        d[i]+=s;
    }
}

int gs(int x)
{
    int ans=0;
    for(int i=x;i;i-=low(i)){
        ans+=d[i];
    }
    return ans;
}
int vis,cnt;
int End[maxn],Next[maxn],Last[maxn];
void dfs(int x)
{
    int i,j,k;
    in[x]=++vis;
    for(i=Last[x];i;i=Next[i]){
        int en=End[i];
        dfs(en);
    }
    ou[x]=vis;
    md(in[x],1);
    md(ou[x],-1);
}
void insert(int x,int y){
    Next[++cnt]=Last[x];
    Last[x]=cnt;
    End[cnt]=y;
}
int tmain()
{

    int i,j,k;
    scanf("%d",&n);
    for(i=1;i<=n-1;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        insert(x,y);
    }
    dfs(1);
    scanf("%d",&m);

    for(i=1;i<=m+n-1;i++){
        char t;
        int x,y;
        cin>>t;
        if(t=='W'){
            scanf("%d",&x);
            printf("%d\n",gs(in[x]-1));
        }
        else {
            scanf("%d%d",&x,&y);
            md(in[y]-1,-1);
            md(ou[y],1);
        }

    }
}
const int main_stack=16;   
char my_stack[128<<20];   
int main() {   
  __asm__("movl %%esp, (%%eax);\n"::"a"(my_astack):"memory");   
  __asm__("movl %%eax, %%esp;\n"::"a"(my_stack+sizeof(my_stack)-main_stack):"%esp");   
  tmain();   
  __asm__("movl (%%eax), %%esp;\n"::"a"(my_stack):"%esp");   
  return 0;   
}