洛谷P4312 [COCI 2009] OTOCI

题目描述

给出n个结点以及每个点初始时对应的权值wi。起始时点与点之间没有连边。有3类操作：

1. bridge A B：询问结点A与结点B是否连通。如果是则输出“no”。否则输出“yes”，并且在结点A和结点B之间连一条无向边。
2. penguins A X：将结点A对应的权值wA修改为X。
3. excursion A B：如果结点A和结点B不连通，则输出“impossible”。否则输出结点A到结点B的路径上的点对应的权值的和。

给出q个操作，要求在线处理所有操作。

输入输出格式

输入格式：

第一行包含一个整数n(1<=n<=30000)，表示节点的数目。

第二行包含n个整数，第i个整数表示第i个节点初始时对应的权值。

第三行包含一个整数q(1<=n<=300000)，表示操作的数目。

以下q行，每行包含一个操作，操作的类别见题目描述。

任意时刻每个节点对应的权值都是1到1000的整数。

输出格式：

输出所有bridge操作和excursion操作对应的输出，每个一行。

输入输出样例

输入样例#1： 

5
4 2 4 5 6
10
excursion 1 1
excursion 1 2
bridge 1 2
excursion 1 2
bridge 3 4
bridge 3 5
excursion 4 5
bridge 1 3
excursion 2 4
excursion 2 5

输出样例#1： 

4
impossible
yes
6
yes
yes
15
yes
15
16

说明

数据范围：1<=n<=30000, 1<=q<=300000, 0<=wi<=1000。

裸的LCT。。。

直接上板子。。。

附代码：

#include
#include
#include
#define MAXN 40010
using namespace std;
int n,m,w[MAXN];
int top,stack[MAXN];
struct node{
	int f,flag,son[2];
	int v;
}a[MAXN];
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
inline bool isroot(int rt){
	return a[a[rt].f].son[0]!=rt&&a[a[rt].f].son[1]!=rt;
}
inline void pushup(int rt){
	if(!rt)return;
	a[rt].v=w[rt]+a[a[rt].son[0]].v+a[a[rt].son[1]].v;
}
inline void pushdown(int rt){
	if(!rt||!a[rt].flag)return;
	a[a[rt].son[0]].flag^=1;a[a[rt].son[1]].flag^=1;a[rt].flag^=1;
	swap(a[rt].son[0],a[rt].son[1]);
}
inline void turn(int rt){
	int x=a[rt].f,y=a[x].f,k=a[x].son[0]==rt?1:0;
	if(!isroot(x)){
		if(a[y].son[0]==x)a[y].son[0]=rt;
		else a[y].son[1]=rt;
	}
	a[rt].f=y;a[x].f=rt;a[a[rt].son[k]].f=x;
	a[x].son[k^1]=a[rt].son[k];a[rt].son[k]=x;
	pushup(x);pushup(rt);
}
void splay(int rt){
	top=0;
	stack[++top]=rt;
	for(int i=rt;!isroot(i);i=a[i].f)stack[++top]=a[i].f;
	while(top)pushdown(stack[top--]);
	while(!isroot(rt)){
		int x=a[rt].f,y=a[x].f;
		if(!isroot(x)){
			if((a[y].son[0]==x)^(a[x].son[0]==rt))turn(rt);
			else turn(x);
		}
		turn(rt);
	}
}
void access(int rt){
	for(int i=0;rt;i=rt,rt=a[rt].f){
		splay(rt);
		a[rt].son[1]=i;
		pushup(rt);
	}
}
inline void makeroot(int rt){access(rt);splay(rt);a[rt].flag^=1;}
int find(int rt){
	access(rt);splay(rt);
	while(a[rt].son[0])rt=a[rt].son[0];
	return rt;
}
inline void split(int x,int y){makeroot(x);access(y);splay(y);}
inline void link(int x,int y){makeroot(x);a[x].f=y;}
inline void cut(int x,int y){split(x,y);a[y].son[0]=a[x].f=0;}
void work(){
	char ch[20];
	int x,y;
	m=read();
	while(m--){
		scanf("%s",ch);x=read();y=read();
		if(ch[0]=='b'){
			int fx=find(x),fy=find(y);
			if(fx==fy)printf("no\n");
			else{
				printf("yes\n");
				link(x,y);
			}
		}
		if(ch[0]=='p'){
			access(x);splay(x);w[x]=y;pushup(x);
		}
		if(ch[0]=='e'){
			int fx=find(x),fy=find(y);
			if(fx!=fy)printf("impossible\n");
			else{
				split(x,y);
				printf("%d\n",a[y].v);
			}
		}
	}
}
void init(){
	n=read();
	for(int i=1;i<=n;i++)a[i].v=w[i]=read();
}
int main(){
	init();
	work();
	return 0;
}