本来是为了刷dfs序的题目的,但是用了树链剖分做。。。
题目输入解释有点问题,但是应该能理解
CODE:
#include<cstdio>
const int N=250005;
const int INF=1e9;
struct edge
{
int nxt,to;
}a[N<<2];
int t[N<<2];
int head[N],size[N],f[N],son[N],pos[N],deep[N],top[N];
int n,m,x,y,num,tot;
inline int max(const int &a,const int &b){return a>b?a:b;}
inline int min(const int &a,const int &b){return a<b?a:b;}
inline void swap(int &a,int &b){a^=b,b^=a,a^=b;}
inline void add(int x,int y)
{
a[++num].nxt=head[x],a[num].to=y,head[x]=num;
a[++num].nxt=head[y],a[num].to=x,head[y]=num;
}
void dfs(int now,int fa,int depth)
{
deep[now]=depth;
f[now]=fa;
size[now]=1;
int tmp=-INF;
for(int i=head[now];i;i=a[i].nxt)
if(a[i].to!=fa)
{
dfs(a[i].to,now,depth+1);
size[now]+=size[a[i].to];
if(size[a[i].to]>tmp) tmp=size[a[i].to],son[now]=a[i].to;
}
}
void dfs2(int now,int high)
{
top[now]=high;
pos[now]=++tot;
if(son[now]) dfs2(son[now],high);
for(int i=head[now];i;i=a[i].nxt)
if(a[i].to!=f[now]&&a[i].to!=son[now]) dfs2(a[i].to,a[i].to);
}
inline void update(int now)
{
t[now]=t[now<<1]+t[now<<1|1];
}
void build(int l,int r,int now)
{
if(l==r)
{
if(l!=1) t[now]=1;
return;
}
int mid=(l+r)>>1;
build(l,mid,now<<1);
build(mid+1,r,now<<1|1);
update(now);
}
void Add(int p,int l,int r,int now)
{
if(l==r)
{
t[now]=0;
return;
}
int mid=(l+r)>>1;
if(p<=mid) Add(p,l,mid,now<<1);
else Add(p,mid+1,r,now<<1|1);
update(now);
}
int ask(int L,int R,int l,int r,int now)
{
if(L<=l&&r<=R) return t[now];
int mid=(l+r)>>1,ans=0;
if(L<=mid) ans=ask(L,R,l,mid,now<<1);
if(R>mid) ans+=ask(L,R,mid+1,r,now<<1|1);
return ans;
}
int askpath(int x,int y)
{
int L,R;
if(top[x]==top[y])
{
L=min(pos[x],pos[y]);
R=max(pos[x],pos[y]);
return ask(L,R,1,n,1);
}
if(deep[top[x]]<deep[top[y]]) swap(x,y);
L=min(pos[x],pos[top[x]]);
R=max(pos[x],pos[top[x]]);
int ans=ask(L,R,1,n,1);
return ans+askpath(y,f[top[x]]);
}
int main()
{
scanf("%d",&n);
for(int i=1;i<n;i++)
scanf("%d%d",&x,&y),add(x,y);
dfs(1,0,1);
dfs2(1,1);
build(1,n,1);
scanf("%d",&m);
m+=n-1;
while(m--)
{
char c=getchar();
while(c!='W'&&c!='A') c=getchar();
if(c=='W') scanf("%d",&x),printf("%d\n",askpath(1,x));
else scanf("%d%d",&x,&y),f[y]==x?Add(pos[y],1,n,1):Add(pos[x],1,n,1);
}
return 0;
}
反思:以后一定要注意树链剖分在修改的时候一定是修改的pos[x]而不是x!