[USACO18FEB]New Barns P,洛谷P4271,LCT维护直径

正题

      比较蠢,没想到.

      其实给两个联通块连一条边也是可以的,就是将两个联通块的直径端点两两求一次距离,求距离可以用LCT来维护点权,然后split一下-1就是距离了.再用一个带权并查集来维护两个直径端点就可以了.

#include<bits/stdc++.h>
using namespace std;

const int N=100010;
struct node{
	int son[2],fa,tot;
	bool swp;
}s[N];
int a[N],b[N],fa[N],q,tot;

#define ls s[x].son[0]
#define rs s[x].son[1]
void upd(int x){s[x].tot=s[ls].tot+s[rs].tot+1;}
void psd(int x){
	if(s[x].swp){
		s[x].swp^=1;s[ls].swp^=1;s[rs].swp^=1;
		swap(s[ls].son[0],s[ls].son[1]);
		swap(s[rs].son[0],s[rs].son[1]);
	}
}
bool isrt(int x){return s[s[x].fa].son[0]!=x && s[s[x].fa].son[1]!=x;}
void rotate(int x){
	int f=s[x].fa,ff=s[f].fa;
	bool w=s[f].son[1]==x;
	if(!isrt(f)) s[ff].son[s[ff].son[1]==f]=x;s[x].fa=ff;
	s[f].son[w]=s[x].son[w^1];s[s[x].son[w^1]].fa=f;
	s[f].fa=x;s[x].son[w^1]=f;
	upd(f);upd(x);
}
void dfs(int x){if(!isrt(x)) dfs(s[x].fa);psd(x);}
void splay(int x){
	dfs(x);
	while(!isrt(x)){
		int f=s[x].fa,ff=s[f].fa;
		if(!isrt(f)) s[ff].son[1]==f^s[f].son[1]==x?rotate(x):rotate(f);
		rotate(x);
	}
}

void acs(int x){for(int las=0;x;las=x,x=s[x].fa) splay(x),s[x].son[1]=las,upd(x);}
void mkrt(int x){acs(x);splay(x);s[x].swp^=1;swap(s[x].son[0],s[x].son[1]);}
void split(int x,int y){mkrt(x);acs(y);splay(y);}
int gl(int x,int y){split(x,y);return s[y].tot-1;}
int findpa(int x){return x!=fa[x]?fa[x]=findpa(fa[x]):x;}
void link(int x,int y){split(x,y);s[x].fa=y;}

int main(){
	scanf("%d",&q);
	char ch[2];
	int x;
	while(q--){
		scanf("%s %d",ch,&x);
		if(ch[0]=='B') {
			tot++;
			if(x==-1) a[tot]=b[tot]=tot,fa[tot]=tot;
			else{
				link(tot,x);fa[tot]=x;
				int now=findpa(tot),A=gl(a[now],b[now]),B=gl(a[now],tot),C=gl(b[now],tot);
				if(B>=A && B>=C) b[now]=tot;
				else if(C>=A && C>=B) a[now]=tot;
			}
		}
		else {
			int now=findpa(x);
			printf("%d\n",max(gl(a[now],x),gl(b[now],x)));
		}
	}
}

猜你喜欢

转载自blog.csdn.net/Deep_Kevin/article/details/108351644