传送门

拿到题目,一看 裸LCT (其实是我懒得打,splay又臭又长)

首先,这道题的意思就是删掉一些边

所以常规操作 点权转边权

之后对于战争操作,在对应的边上+1

对于和平操作,在对应的边上-1

然后对于询问,即统计从\(u\)到\(v\)的路径上的和

如果和为\(0\),即表明可以到,否则不能到

代码


#include<bits/stdc++.h>
namespace my_std
{
using namespace std;
#define rep(i,a,b) for (int i=(a);i<=(b);i++)
#define drep(i,a,b) for (int i=(a);i>=(b);i--)
#define go(u) for (int i=head[(u)];i;i=e[i].nxt)
#define pf printf
#define writeln(x) write(x),putchar('\n')
#define writesp(x) write(x),putchar(' ')
#define mem(x,v) memset(x,v,sizeof(x))
typedef long long ll;
const int INF=0x7fffffff;
inline int read()
{
int sum=0,f=1;
char c=0;
while (!isdigit(c))
{
if (c=='-') f=-1;
c=getchar();
}
while (isdigit(c))
{
sum=(sum<<1)+(sum<<3)+(c^48);
c=getchar();
}
return sum*f;
}
void write(int k)
{
if (k<0) putchar('-'),k=-k;
if (k>=10) write(k/10);
putchar(k%10+'0');
}
}
using namespace my_std;
const int N=300010;
int n,quiz,cnt,tree[N],head[N];
int son[N],top[N],size[N];
int idx,in[N],num[N],f[N],dep[N];
struct edge
{
int to,nxt;
}e[N<<1];
struct node
{
int x,y;
}a[N];
#define lowbit(x) (x&(-x))
void update(int x,int y)
{
while (x<=n)
{
tree[x]+=y;
x+=lowbit(x);
}
}
int getsum(int x)
{
int res=0;
while (x>0)
{
res+=tree[x];
x-=lowbit(x);
}
return res;
}
int query(int l,int r)
{
return getsum(r)-getsum(l-1);
}
void add(int u,int v)
{
e[++cnt].to=v;
e[cnt].nxt=head[u];
head[u]=cnt;
}
void dfs1(int u,int fa)
{
dep[u]=dep[fa]+1;
f[u]=fa;
size[u]=1;
go(u)
{
int v=e[i].to;
if (v==fa) continue;
dfs1(v,u);
size[u]+=size[v];
if (size[son[u]]<size[v])
{
son[u]=v;
}
}
}
void dfs2(int u,int t)
{
in[u]=++idx;
num[idx]=u;
top[u]=t;
if (son[u]==0) return;
dfs2(son[u],t);
go(u)
{
int v=e[i].to;
if (v==f[u]||v==son[u]) continue;
dfs2(v,v);
}
}
int query_tree(int x,int y)
{
int res=0;
while (top[x]!=top[y])
{
if (dep[top[x]]<dep[top[y]]) swap(x,y);
res+=query(in[top[x]],in[x]);
x=f[top[x]];
}
if (dep[x]<dep[y]) swap(x,y);
res+=query(in[y]+1,in[x]);
return res;
}
int main()
{
n=read(),quiz=read();
int u,v;
rep(i,1,n-1)
{
u=read(),v=read();
add(u,v),add(v,u);
}
dfs1(1,0);
dfs2(1,1);
char opt;
int p,q;
cnt=0;
rep(i,1,quiz)
{
scanf("%s",&opt);
if (opt=='Q')
{
p=read(),q=read();
int res=query_tree(p,q);
if (res==0) pf ("Yes\n");
else pf ("No\n");
}
if (opt=='C')
{
p=read(),q=read();
if (dep[p]<dep[q]) swap(p,q);
update(in[p],1);
a[++cnt].x=p;
a[cnt].y=q;
}
if (opt=='U')
{
p=read();
update(in[a[p].x],-1);
}
}
}
05-19 03:43