SP16549 QTREE6 - Query on a tree VI(LCT)

时间:2023-03-09 17:47:23
SP16549 QTREE6 - Query on a tree VI(LCT)

题意翻译

题目描述

给你一棵n个点的树,编号1~n。每个点可以是黑色,可以是白色。初始时所有点都是黑色。下面有两种操作请你操作给我们看:

0 u:询问有多少个节点v满足路径u到v上所有节点(包括)都拥有相同的颜色
1 u:翻转u的颜色

输入格式

一行一个整数n

接下来n-1行,每行两个整数表示一条边

接下来一行一个整数m表示操作次数

接下来m行,每行两个整数分别表示操作类型和被操作节点 输出格式

对每个询问操作输出相应的结果

题解

简单来说,就是维护同色联通块的大小

干脆直接暴力linkcut好了(T上天)

然后来介绍一下(刚学会的)一种维护染色联通块的较为常用的模型

很多与树有关的题目,当边权不好处理时,可以把边权转化为儿子的点权来做

然后这里恰恰相反,要把点权给转化为边权

我们把每一个节点的颜色赋给与它父亲相连的边

弄两个LCT,对应两种颜色,一种颜色的边只有在对应的LCT中才会相连

于是,同色的联通块,就转化为了减去顶部节点后的边的连通块(因为顶部节点在这个LCT是没有向上连边的,说明顶部节点并不是这个颜色,那么他的所有子树并不连通,要断开才行)

然后可以发现,修改点的颜色之后,只要在原来的LCT中cut,在新的LCT上link就行啦

查询怎么做呢?先access,然后findroot,再输出root的实子树大小就行了

ps:1本来没有父亲,但为了方便,可以建一个虚点n+1,令他作为1的父亲就好了

 //minamoto
#include<iostream>
#include<cstdio>
#define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[<<],*p1=buf,*p2=buf;
inline int read(){
#define num ch-'0'
char ch;bool flag=;int res;
while(!isdigit(ch=getc()))
(ch=='-')&&(flag=true);
for(res=num;isdigit(ch=getc());res=res*+num);
(flag)&&(res=-res);
#undef num
return res;
}
char sr[<<],z[];int C=-,Z;
inline void Ot(){fwrite(sr,,C+,stdout),C=-;}
inline void print(int x){
if(C><<)Ot();if(x<)sr[++C]=,x=-x;
while(z[++Z]=x%+,x/=);
while(sr[++C]=z[Z],--Z);sr[++C]='\n';
}
const int N=;
int f[N],Next[N<<],head[N],ver[N<<],tot,col[N],n,m;
inline void add(int u,int v){
ver[++tot]=v,Next[tot]=head[u],head[u]=tot;
ver[++tot]=u,Next[tot]=head[v],head[v]=tot;
}
struct LCT{
int fa[N],ch[N][],si[N],sum[N];
LCT(){for(int i=;i<=n+;++i) sum[i]=;}
inline bool isroot(int x){return ch[fa[x]][]!=x&&ch[fa[x]][]!=x;}
#define lc ch[x][0]
#define rc ch[x][1]
inline void pushup(int x){sum[x]=sum[lc]+sum[rc]+si[x]+;}
void rotate(int x){
int y=fa[x],z=fa[y],d=ch[y][]==x;
if(!isroot(y)) ch[z][ch[z][]==y]=x;
fa[x]=z,fa[y]=x,fa[ch[x][d^]]=y,ch[y][d]=ch[x][d^],ch[x][d^]=y,pushup(y);
}
void splay(int x){
for(int y=fa[x],z=fa[y];!isroot(x);y=fa[x],z=fa[y]){
if(!isroot(y))
((ch[y][]==x)^(ch[z][]==y))?rotate(x):rotate(y);
rotate(x);
}
pushup(x);
}
void access(int x){
for(int y=;x;x=fa[y=x]){
splay(x);
si[x]+=sum[rc];
si[x]-=sum[rc=y];
}
}
int findroot(int x){
access(x),splay(x);
while(lc) x=lc;
splay(x);
return x;
}
void link(int x){
access(x),splay(x);
int y=fa[x]=f[x];
access(y),splay(y);
si[y]+=sum[x],sum[y]+=sum[x];
}
void cut(int x){
access(x),splay(x);
lc=fa[lc]=;
pushup(x);
}
}lct[];
void dfs(int u){
for(int i=head[u];i;i=Next[i]){
int v=ver[i];
if(v==f[u]) continue;
f[v]=u,dfs(v),lct[].link(v);
}
}
int main(){
//freopen("testdata.in","r",stdin);
n=read();
for(int i=;i<n;++i){
int u=read(),v=read();
add(u,v);
}
dfs();
f[]=n+,lct[].link();
m=read();
while(m--){
int op=read(),u=read();
if(op) lct[col[u]].cut(u),lct[col[u]^=].link(u);
else{
int v=lct[col[u]].findroot(u);
print(lct[col[u]].sum[lct[col[u]].ch[v][]]);
}
}
Ot();
return ;
}