1. 程式人生 > >[BZOJ1103][POI2007]大都市meg

[BZOJ1103][POI2007]大都市meg

Time Limit: 10 Sec
Memory Limit: 162 MB

Description

在經濟全球化浪潮的影響下,習慣於漫步在清晨的鄉間小路的郵遞員Blue Mary也開始騎著摩托車傳遞郵件了。不過,她經常回憶起以前在鄉間漫步的情景。昔日,鄉下有依次編號為1…n的n個小村莊,某些村莊之間有一些雙向的土路。從每個村莊都恰好有一條路徑到達村莊1(即位元堡)。並且,對於每個村莊,它到位元堡的路徑恰好只經過編號比它的編號小的村莊。另外,對於所有道路而言,它們都不在除村莊以外的其他地點相遇。在這個未開化的地方,從來沒有過高架橋和地下鐵道。隨著時間的推移,越來越多的土路被改造成了公路。至今,Blue Mary還清晰地記得最後一條土路被改造為公路的情景。現在,這裡已經沒有土路了——所有的路都成為了公路,而昔日的村莊已經變成了一個大都市。 Blue Mary想起了在改造期間她送信的經歷。她從位元堡出發,需要去某個村莊,並且在兩次送信經歷的間隔期間,有某些土路被改造成了公路.現在Blue Mary需要你的幫助:計算出每次送信她需要走過的土路數目。(對於公路,她可以騎摩托車;而對於土路,她就只好推車了。)

Input

第一行是一個數 n ( 1 < = n < = 250000

) n(1 < = n < = 2 50000) .以下 n 1 n-1 行,每行兩個整 a
b 1 < = a < = b ) a,b(1 < = a<=b)
以下一行包含一個整數 m ( 1 < = m < = 250000 ) m(1 < = m < = 2 50000) ,表示Blue Mary曾經在改造期間送過 m m 次信。以下 n + m 1 n+m-1 行,每行有兩種格式的若干資訊,表示按時間先後發生過的 n + m 1 n+m-1 次事件:若這行為 A a b ( a < = b ) (a<=b) 表示將從a到b的土路改成公路。
若這行為 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
在這裡插入圖片描述

題解:
dfs序+線段樹區間修改。
修路操作當成子樹全部結點權值 1 -1
查詢的話就是單點查詢了。

#include<bits/stdc++.h>
#define LiangJiaJun main
using namespace std;

struct seg{
    int l,r,w,tag;
}tr[1000004];
struct edge{
    int to,nt;
}e[500004];
int cnt;
int n,m,ne,h[250004];
int dfn[250004],kcf[250004],lp[250004],opg[250004];
void add(int u,int v){
     e[++ne].to=v;e[ne].nt=h[u];
     h[u]=ne;
}
void dfs(int x){
     dfn[x]=++cnt;
     kcf[cnt]=x;
     for(int i=h[x];i;i=e[i].nt){
         if(!dfn[e[i].to]){
            opg[e[i].to]=opg[x]+1;
            dfs(e[i].to);
         }
     }
     lp[x]=cnt;
}
void push(int k){
     int x=tr[k].tag;
     tr[k<<1].tag+=x;
     tr[k<<1|1].tag+=x;
     tr[k].tag=0;
     tr[k<<1].w+=x*(tr[k<<1].r-tr[k<<1].l+1);
     tr[k<<1|1].w+=x*(tr[k<<1|1].r-tr[k<<1|1].l+1);
     return ;
}
void build(int k,int l,int r){
     tr[k].l=l;tr[k].r=r;
     tr[k].tag=0;
     if(l==r){
        tr[k].w=opg[kcf[l]];
        return ;
     }
     int mid=(l+r)>>1;
     build(k<<1,l,mid);
     build(k<<1|1,mid+1,r);
     tr[k].w=tr[k<<1].w+tr[k<<1|1].w;
}
void change(int k,int a,int b,int x){
     int l=tr[k].l,r=tr[k].r;
     if(l==a&&r==b){
        tr[k].w+=x*(r-l+1);
        tr[k].tag+=x;
        return ;
     }
     if(tr[k].tag)push(k);
     int mid=(l+r)>>1;
     if(b<=mid)change(k<<1,a,b,x);
     else if(a>mid)change(k<<1|1,a,b,x);
     else{
        change(k<<1,a,mid,x);
        change(k<<1|1,mid+1,b,x);
     }
     tr[k].w=tr[k<<1].w+tr[k<<1|1].w;
}
int ask(int k,int a){
    int l=tr[k].l,r=tr[k].r;
    if(l==a&&r==a){
        return tr[k].w;
    }
    if(tr[k].tag)push(k);
    int mid=(l+r)>>1;
    if(a<=mid)return ask(k<<1,a);
    else return ask(k<<1|1,a);
}
int w33ha(){
    ne=0;
    memset(h,0,sizeof(h));
    for(int i=1;i<=n;i++){
        dfn[i]=0;
        lp[i]=0;
        kcf[i]=0;
    }
    cnt=0;
    opg[1]=0;
    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    dfs(1);
    build(1,1,cnt);
    scanf("%d",&m);
    m=n+m-1;
    while(m--){
        char ch[4];
        int x,y;
        scanf("%s",ch+1);
        if(ch[1]=='A'){
            scanf("%d%d",&x,&y);
            change(1,dfn[max(x,y)],lp[max(x,y)],-1);
        }
        if(ch[1]=='W'){
            scanf("%d",&x);
            printf("%d\n",ask(1,dfn[x]));
        }
    }
    return 0;
}

int LiangJiaJun(){
    while(scanf("%d",&n)!=EOF)w33ha();
    return 0;
}