【BZOJ】1180: [CROATIAN2009]OTOCI（lct）

#include <cstdio>
#include <cstring>
#include <cmath>
#include <string>
#include <iostream>
#include <algorithm>
#include <queue>
#include <set>
#include <map>
using namespace std;
typedef long long ll;
#define rep(i, n) for(int i=0; i<(n); ++i)
#define for1(i,a,n) for(int i=(a);i<=(n);++i)
#define for2(i,a,n) for(int i=(a);i<(n);++i)
#define for3(i,a,n) for(int i=(a);i>=(n);--i)
#define for4(i,a,n) for(int i=(a);i>(n);--i)
#define CC(i,a) memset(i,a,sizeof(i))
#define read(a) a=getint()
#define print(a) printf("%d", a)
#define dbg(x) cout << (#x) << " = " << (x) << endl
#define error(x) (!(x)?puts("error"):0)
#define rdm(x, i) for(int i=ihead[x]; i; i=e[i].next)
inline const int getint() { int r=0, k=1; char c=getchar(); for(; c<‘0‘||c>‘9‘; c=getchar()) if(c==‘-‘) k=-1; for(; c>=‘0‘&&c<=‘9‘; c=getchar()) r=r*10+c-‘0‘; return k*r; }

struct node* null;
struct node {
	node *f, *c[2];
	int w, s; bool rev;
	node(int _k=0) { w=s=_k; f=c[0]=c[1]=null; rev=0; }
	void setc(node *x, bool d) { c[d]=x; x->f=this; }
	bool check() { return f==null || (f->c[0]!=this && f->c[1]!=this); }
	bool d() { return f->c[1]==this; }
	void pushup() { s=w+c[0]->s+c[1]->s; }
	void upd() { if(this==null) return; rev=!rev; swap(c[0], c[1]); }
	void pushdown() {
		if(rev) {
			rev=0;
			c[0]->upd();
			c[1]->upd();
		}
	}
};
void rot(node *x) {
	node *f=x->f;
	f->pushdown(); x->pushdown(); bool d=x->d();
	if(f->check()) x->f=f->f;
	else f->f->setc(x, f->d());
	f->setc(x->c[!d], d);
	x->setc(f, !d);
	f->pushup();
}
void fix(node *x) { if(!x->check()) fix(x->f); x->pushdown(); }
void splay(node *x) {
	fix(x);
	while(!x->check())
		if(x->f->check()) rot(x);
		else x->d()==x->f->d()?(rot(x->f), rot(x)):(rot(x), rot(x));
	x->pushup();
}
node* access(node *x) {
	node *y=null;
	for(; x!=null; y=x, x=x->f) splay(x), x->c[1]=y;
	return y;
}
node* findroot(node *x) {
	access(x); splay(x);
	while(x->c[0]!=null) x=x->c[0];
	return x;
}
void mkroot(node *x) { access(x)->upd(); splay(x); }
void link(node *x, node *y) { mkroot(x); x->f=y; }

const int N=30005;
node *root[N];
int n;
int main() {
	null=new node; null->f=null->c[0]=null->c[1]=null;
	read(n);
	for1(i, 1, n) root[i]=new node(getint());
	int m=getint();
	while(m--) {
		char c=getchar(); while(c<‘a‘||c>‘z‘) c=getchar();
		int a=getint(), b=getint();
		if(c==‘b‘) {
			if(findroot(root[a])==findroot(root[b])) puts("no");
			else { puts("yes"); link(root[a], root[b]); };
		}
		else if(c==‘p‘) { mkroot(root[a]); root[a]->w=b; root[a]->pushup(); }
		else {
			if(findroot(root[a])!=findroot(root[b])) { puts("impossible"); continue; }
			mkroot(root[a]); access(root[b]); splay(root[b]);
			printf("%d\n", root[b]->s);
		}
	}
	return 0;
}