[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
第一行是一個數
.以下
行,每行兩個整
以下一行包含一個整數
,表示Blue Mary曾經在改造期間送過
次信。以下
行,每行有兩種格式的若干資訊,表示按時間先後發生過的
次事件:若這行為 A 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序+線段樹區間修改。
修路操作當成子樹全部結點權值
查詢的話就是單點查詢了。
#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;
}