1103: [POI2007]大都市meg
Time Limit: 10 Sec Memory Limit: 162 MBSubmit: 3012 Solved: 1581
[Submit][Status][Discuss]
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
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
1
0
1
HINT
Source
可以用很裸的树链剖分来写,也可以dfs序+线段树 我写的是树链剖分
#include <bits/stdc++.h> using namespace std; inline int read(){ int x=0;int f=1;char ch=getchar(); while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();} while(isdigit(ch)) {x=x*10+ch-'0';ch=getchar();} return x*f; } const int MAXN=1e6+10; struct sigment_tree{ int sum; }T[MAXN]; namespace zhangenming{ struct node{ int y,next; }e[MAXN<<1]; int linkk[MAXN<<1],c[MAXN<<1]={},siz[MAXN]={},fa[MAXN],dep[MAXN]={},son[MAXN],Node[MAXN],cnt[MAXN]={},top[MAXN]={},tot=0,m,len=0,n,x,y; void insert(int xx,int yy){ e[++len].y=yy;e[len].next=linkk[xx];linkk[xx]=len; } void dfs1(int father,int node,int depth){ siz[node]=1;fa[node]=father;dep[node]=depth;son[node]=0;int maxxn=0; for(int i=linkk[node];i;i=e[i].next){ if(e[i].y!=father){ dfs1(node,e[i].y,depth+1); if(siz[e[i].y]>maxxn) son[node]=e[i].y,maxxn=siz[e[i].y]; siz[node]+=siz[e[i].y]; } } //cout<<siz[node]<<endl; } void dfs2(int node,int id){ cnt[node]=++tot;Node[tot]=node;top[node]=id; if(son[node]) dfs2(son[node],id); for(int i=linkk[node];i;i=e[i].next){ if(e[i].y!=fa[node]&&e[i].y!=son[node]) dfs2(e[i].y,e[i].y); } } void buildtree(int l,int r,int root){ if(l==r) {T[root].sum=1;if(l==1) T[root].sum=0;return;} int mid=(l+r)>>1; buildtree(l,mid,root<<1); buildtree(mid+1,r,root<<1|1); T[root].sum=T[root<<1].sum+T[root<<1|1].sum; } void insertt(int l,int r,int root){ if(l>x||r<x) return; if(l==r) {T[root].sum=0;return;} int mid=(l+r)>>1; insertt(l,mid,root<<1); insertt(mid+1,r,root<<1|1); T[root].sum=T[root<<1].sum+T[root<<1|1].sum; } int find(int l,int r,int root){ if(l>=x&&r<=y) return T[root].sum; if(l>y||r<x) return 0; int mid=(r+l)>>1; return find(l,mid,root<<1)+find(mid+1,r,root<<1|1); } void init(){ n=read(); for(int i=1;i<n;i++){ int xx=read();int yy=read(); insert(xx,yy); insert(yy,xx); } dfs1(0,1,0); dfs2(1,1); buildtree(1,n,1); } int search(int node){ if(node==1) return 0; int ans=0; while(top[node]!=1){ x=cnt[top[node]]; y=cnt[node]; ans+=find(1,n,1); node=fa[top[node]]; } if(node==1) return ans; y=cnt[node]; x=cnt[top[node]]; ans+=find(1,n,1); return ans; } void solve(){ m=read()+n-1; while(m--){ char ch; scanf("%c",&ch); if(ch=='A'){ int xx=read();int yy=read(); if(fa[xx]==yy) swap(xx,yy); x=cnt[yy]; insertt(1,n,1); } if(ch=='W'){ int xx=read(); printf("%d ",search(xx)); } } } } int main(){ using namespace zhangenming; init(); /*for(int i=1;i<=n;i++){ cout<<cnt[i]<<' '<<top[i]<<' '<<fa[i]<<' '<<son[i]<<' '<<Node[i]<<' '<<siz[i]<<' '<<endl; }*/ solve(); return 0; }