[bzoj 3224] 일반 밸 런 스 트 리

Description
질서정연 한 수열 을 유지 하기 위해 서 는 데이터 구조 (제목 제목 참조) 를 써 야 합 니 다. 다음 과 같은 동작 을 제공 해 야 합 니 다. 예 를 들 어 기 존 서열 은 5 입 니 다. 4 3 2 1, 뒤 집기 구간 이 [2, 4] 라면 결 과 는 5. 2 3 4 1 
Input
첫 번 째 행동 n, m n 은 초기 서열 에 n 개의 수가 있 음 을 나타 낸다. 이 서열 은 순서대로 (1, 2... n - 1, n)  m 는 뒤 집기 동작 횟수 를 나타 내 고 다음 m 줄 마다 두 개의 수 [l, r] 를 나타 낸다. 데이터 보증 1<=l<=r<=n 
Output
 
n 개의 숫자 를 출력 하여 원시 서열 이 m 차 변환 을 거 친 결 과 를 나타 낸다. 
Sample Input
5 3 1 3 1 3 1 4
Sample Output
4 3 2 1 5
HINT
N,M<=100000
Source
밸 런 스 트 리
[문제 풀이]
하루 종일 널 빤 지 를 쳤 다.
【 코드 】
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define MAXN 1000000
int ch[MAXN][2],f[MAXN],size[MAXN],cnt[MAXN],key[MAXN];
int sz,root;
inline void clear(int x){
	ch[x][0]=ch[x][1]=f[x]=size[x]=cnt[x]=key[x]=0;
}
inline bool get(int x){
	return ch[f[x]][1]==x;
}
inline void update(int x){
	if (x){
		size[x]=cnt[x];
		if (ch[x][0]) size[x]+=size[ch[x][0]];
		if (ch[x][1]) size[x]+=size[ch[x][1]];
	}
}
inline void rotate(int x){
	int old=f[x],oldf=f[old],whichx=get(x);
	ch[old][whichx]=ch[x][whichx^1]; f[ch[old][whichx]]=old;
	ch[x][whichx^1]=old; f[old]=x;
	f[x]=oldf;
	if (oldf)
		ch[oldf][ch[oldf][1]==old]=x;
	update(old); update(x);
}
inline void splay(int x){
	for (int fa;fa=f[x];rotate(x))
	  if (f[fa])
	    rotate((get(x)==get(fa))?fa:x);
	root=x;
}
inline void insert(int x){
	if (root==0){sz++; ch[sz][0]=ch[sz][1]=f[sz]=0; root=sz; size[sz]=cnt[sz]=1; key[sz]=x; return;}
	int now=root,fa=0;
	while(1){
		if (x==key[now]){
			cnt[now]++; update(now); update(fa); splay(now); break;
		}
		fa=now;
		now=ch[now][key[now]<x];
		if (now==0){
			sz++;
			ch[sz][0]=ch[sz][1]=0;
			f[sz]=fa;
			size[sz]=cnt[sz]=1;
			ch[fa][key[fa]<x]=sz;
			key[sz]=x;
			update(fa);
			splay(sz);
			break;
		}
	}
}
inline int find(int x){
	int now=root,ans=0;
	while(1){
		if (x<key[now])
		  now=ch[now][0];
		else{
			ans+=(ch[now][0]?size[ch[now][0]]:0);
			if (x==key[now]){
				splay(now); return ans+1;
			}
			ans+=cnt[now];
			now=ch[now][1];
		}
	}
}
inline int findx(int x){
	int now=root;
	while(1){
		if (ch[now][0]&&x<=size[ch[now][0]])
		  now=ch[now][0];
		else{
			int temp=(ch[now][0]?size[ch[now][0]]:0)+cnt[now];
			if (x<=temp) return key[now];
			x-=temp; now=ch[now][1];
		}
	}
}
inline int pre(){
	int now=ch[root][0];
	while (ch[now][1]) now=ch[now][1];
	return now;
}
inline int next(){
	int now=ch[root][1];
	while (ch[now][0]) now=ch[now][0];
	return now;
}
inline void del(int x){
	int whatever=find(x);
	if (cnt[root]>1){cnt[root]--; update(root); return;}
	if (!ch[root][0]&&!ch[root][1]) {clear(root); root=0; return;}
	if (!ch[root][0]){
		int oldroot=root; root=ch[root][1]; f[root]=0; clear(oldroot); return;
	}
	else if (!ch[root][1]){
		int oldroot=root; root=ch[root][0]; f[root]=0; clear(oldroot); return;
	}
	int leftbig=pre(),oldroot=root;
	splay(leftbig);
	ch[root][1]=ch[oldroot][1];
	f[ch[oldroot][1]]=root;
	clear(oldroot);
	update(root); 
}
int main(){
	int n,opt,x;
	scanf("%d",&n);
	for (int i=1;i<=n;++i){
		scanf("%d%d",&opt,&x);
		switch(opt){
			case 1: insert(x); break;
			case 2: del(x); break;
			case 3: printf("%d
",find(x)); break; case 4: printf("%d
",findx(x)); break; case 5: insert(x); printf("%d
",key[pre()]); del(x); break; case 6: insert(x); printf("%d
",key[next()]); del(x); break; } } }

좋은 웹페이지 즐겨찾기