题目描述
Solution
这道题考试的时候竟然没有仔细想,结果只拿了暴力分...
其实就是一个 DFS序+树状数组。
我们先把用 DFS 把它变成一个序列,同时记录它们的 \(siz\)。
那么我们每一次连一条边之后就是对它的子树产生影响。
在树状数组里面维护就好了。
代码
#include<bits/stdc++.h>
using namespace std;
const int maxn=250008;
struct sj{
int to;
int next;
}a[maxn*2];
int head[maxn],size;
int n,m;
void add1(int x,int y)
{
a[++size].to=y;
a[size].next=head[x];
head[x]=size;
}
int id[maxn],tot;
int siz[maxn],ans[maxn];
void dfs(int x)
{
id[x]=++tot;
siz[x]=1;
for(int i=head[x];i;i=a[i].next)
{
int tt=a[i].to;
if(!siz[tt])
{
ans[tt]=ans[x]+1;
dfs(tt);
siz[x]+=siz[tt];
}
}
}
int c[maxn];
int lowbit(int x)
{return x&(-x);}
void add(int x,int z)
{for(int i=x;i<=n;i+=lowbit(i))c[i]+=z;}
int check(int x)
{
int sum=0;
for(int i=x;i>=1;i-=lowbit(i))
sum+=c[i];
return sum;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<n;i++)
{int x,y; scanf("%d%d",&x,&y);add1(x,y); add1(y,x);}
dfs(1);
for(int i=1;i<=n;i++)
add(id[i],0);
scanf("%d",&m);
for(int i=1;i<=m+n-1;i++)
{
int x,y;char ch; cin>>ch;
if(ch=='A')
{
scanf("%d%d",&x,&y);
if(id[x]<id[y])swap(y,x);
add(id[x],1);
add(id[x]+siz[x],-1);
}
if(ch=='W')
scanf("%d",&x),
cout<<ans[x]-check(id[x])<<endl;
}
}
原文地址:https://www.cnblogs.com/Kv-Stalin/p/9313978.html
时间: 2024-10-15 14:43:16