[CF1137F][LCT][樹狀數組]Matches Are Not a Child's Play

CF1137F

LCT神題。。。

首先三操作顯然就是兩個二操作
然後我們考慮一操作會造成什麼影響
我們將權值最大的那個點看作根,一個點進行一操作後,他到根節點這條路徑就會最後被刪除,其他的點刪除順序不變,這很顯然
考慮這段鏈,一定是從原來的最大值那一頭刪到當前最大值這一頭,那我們就把新的最大值作爲根,這就很像LCT了
所以我們把會從下到上按順序刪除的鏈看成實鏈,那麼不同鏈之間的影響只與鏈上最大值有關,那麼可以用權值樹狀數組維護大小關係,然後查詢就是鏈上最大值小於這條鏈的所有鏈長的和減去這個點在splay作爲根時候的左子樹siz(即以當前最大值爲根時,這條鏈下方的點加上其他會在它之前刪除的點)

Code:

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=2e5+5;
namespace BIT{
	int tr[N<<1],n;
	inline int lb(int x){return x&-x;}
	inline void add(int x,int y){for(int i=x;i<=n;i+=lb(i)) tr[i]+=y;}
	inline int ask(int x){int res=0;for(int i=x;i;i-=lb(i)) res+=tr[i];return res;}
}
int cur;
namespace LCT{
	int ls[N],rs[N],fa[N],rev[N],siz[N],val[N];
	inline int isrs(int x){return x==rs[fa[x]];}
	inline bool isroot(int x){
		if(!fa[x]) return true;
		return x!=ls[fa[x]] && x!=rs[fa[x]];
	}
	inline void pushup(int x){siz[x]=siz[ls[x]]+siz[rs[x]]+1;}
	inline void pushdown(int x){
		if(rev[x]){
			swap(ls[x],rs[x]);
			if(ls[x]) rev[ls[x]]^=1;
			if(rs[x]) rev[rs[x]]^=1;
			rev[x]=0;
		}
		if(ls[x]) val[ls[x]]=val[x];
		if(rs[x]) val[rs[x]]=val[x];
	}
	inline void rotate(int x){
		int y=fa[x],z=fa[y],b=(ls[y]==x?rs[x]:ls[x]);
		if(z && !isroot(y)) (ls[z]==y?ls[z]:rs[z])=x;
		fa[x]=z;fa[y]=x;b?fa[b]=y:0;
		if(ls[y]==x) rs[x]=y,ls[y]=b;
		else ls[x]=y,rs[y]=b;
		pushup(y);pushup(x);
	}
	int que[N];
	inline void splay(int x){
		que[que[0]=1]=x;
		for(int y=x;!isroot(y);y=fa[y]) que[++que[0]]=fa[y];
		for(int i=que[0];i;i--) pushdown(que[i]);
		while(!isroot(x)){
			if(!isroot(fa[x])){
				if(isrs(x)==isrs(fa[x])) rotate(fa[x]);
				else rotate(x);
			}
			rotate(x);
		}
	}
	inline void access(int x){
		for(int y=0;x;y=x,x=fa[x]){
			splay(x);
			rs[x]=0;pushup(x);
			BIT::add(val[x],-siz[x]);
			BIT::add(cur,siz[x]);
			rs[x]=y;pushup(x);
		}
	}
	inline void makeroot(int x){++cur;access(x);splay(x);rev[x]^=1;val[x]=cur;}
	inline int ask(int x){splay(x);return BIT::ask(val[x])-siz[ls[x]];}
}
using namespace LCT;
int vis[N<<1],head[N],nxt[N<<1],tot=0;
inline void add(int x,int y){vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;}
void dfs(int v){
	siz[v]=1;val[v]=v;
	for(int i=head[v];i;i=nxt[i]){
		int y=vis[i];
		if(y==fa[v]) continue;
		fa[y]=v;dfs(y);
		if(val[y]>val[v]){
			val[v]=val[y];
			siz[v]=siz[y]+1;
			rs[v]=y;
		}
	}
	BIT::add(val[v],1);
}
int main(){
	int n=read(),m=read();
	cur=n;BIT::n=n+m;
	for(int x,y,i=1;i<n;i++){
		x=read(),y=read();
		add(x,y);add(y,x);
	}
	dfs(n);
	char ch[10];
	while(m--){
		int x,y;
		scanf("%s",ch+1);
		if(ch[1]=='u') x=read(),makeroot(x);
		else if(ch[1]=='w') x=read(),cout<<ask(x)<<"\n";
		else x=read(),y=read(),cout<<(ask(x)<ask(y)?x:y)<<"\n";
	}
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章