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

来源:互联网 发布:怎么样做淘宝客赚钱 编辑:程序博客网 时间:2024/05/17 01:03

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

  认真写博客了,就算题水博客也不能再水了。  将a-b的路径改为公路可以视为将a-b的路径的权值改为0,则题目转为经一系列边权的修改后询问根节点与一个节点的距离。刷一遍DFS序,比如样例中树的dfs序为1223345541。将进入的权值赋为1,离开的权值赋为-1,起点的进出权值都为0(进出权值记录在中间一行)。得:  1  2  2  3  3  4  5  5  4  1  0  1 -1  1 -1  1  1 -1 -1  0  0  1  0  1  0  1  2  1  0  0  然后有什么用?我们可以发现起点到任意点N的距离都为N的进入点的前缀和(最下面一行)。  再试试修改,将1-4的权值改为0,只需将深度较深的点的进入权值改为0,离开的权值也改为0就好了。  1  2  2  3  3  4  5  5  4  1  0  1 -1  1 -1  0  1 -1  0  0  0  1  0  1  0  0  1  0  0  0  显然这类操作我们可以用树状数组直接秒了。  效率(n+m)log(n)。
#include <cstdio>using namespace std;inline char tc(void){    static char fl[1000000],*A=fl,*B=fl;    return A==B&&(B=(A=fl)+fread(fl,1,1000000,stdin),A==B)?EOF:*A++;}inline int read(void){    int a=0;static char c;    while((c=tc())<'0'||c>'9');    while(c>='0'&&c<='9')        a=a*10+c-'0',c=tc();    return a;}inline char Readchar(void){    static char c;    while((c=tc())!='A'&&c!='W');    return c;}inline int min(int a,int b){    return a<b?a:b;}inline int max(int a,int b){    return a>b?a:b;}struct Edge{    int to;    Edge *next;}*h[250001],edge[500001];int n,cnt,tot,l[250001],r[250001],BIT[500001],o,m;inline void insert(int x,int y){    edge[++tot].to=y,edge[tot].next=h[x],h[x]=&edge[tot];return ;}inline void add(int x,int y){    for (;x<=o;x+=x&-x)        BIT[x]+=y;    return ;}inline int sum(int x){    int sum=0;    for (;x;x-=x&-x)        sum+=BIT[x];    return sum;}void dfs(int x,int f){    l[x]=++cnt;if(x!=1) add(l[x],1);    for (Edge *i=h[x];i;i=i->next)        if(i->to!=f)            dfs(i->to,x);    r[x]=++cnt;if(x!=1) add(r[x],-1);    return ;}int main(void){    register int i,x,y;    n=read(),o=2*n;    for (i=1;i<n;++i)        x=read(),y=read(),insert(x,y),insert(y,x);    dfs(1,0),m=read();    for (i=1;i<=n+m-1;++i)        if(Readchar()=='A')            x=read(),y=read(),add(max(l[x],l[y]),-1),add(min(r[x],r[y]),1);        else             x=read(),printf("%d\n",sum(l[x]));    return 0;}