BZOJ 3685 普通van Emde Boas树

题目链接：233333

一棵裸裸的线段树

rank为一个点在现在里面的排名

第一个和第二个操作脑补成单点修改

第三个和第四个操作脑补成看rank求点

第五个和第六个操作脑补成求一个点的rank数然后+1或-1转为看rank求点

第七个操作===你不会我也没办法

程序中的rank可以理解为sum，自己手贱没办法

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#define rep(j,k,l) for (int j=k;j<=l;j++)
#define N 1000005

using namespace std;
struct _233{
	
	int l,r,rank;
	
} tr[N*5];
int a[N],cnt;

int read(){
	
	char ch=getchar();
	int s=0;
	while(ch>'9'||ch<'0') ch=getchar();
	while(ch<='9'&&ch>='0'){
		
		s=s*10+ch-'0';ch=getchar();
		
	}
	return s;
	
}

void stree(int k,int l,int r){
	
	if (l==r) return;
	tr[k].l=++cnt;
	tr[k].r=++cnt;
	stree(tr[k].l,l,(l+r)/2);
	stree(tr[k].r,(l+r)/2+1,r);
	
}

void add(int k,int l,int r,int p){
	
	if (l==r){
		
		tr[k].rank=1;
		return;
		
	}
	int mid=(l+r)/2;
	if (p<=mid) add(tr[k].l,l,(l+r)/2,p);
	else add(tr[k].r,(l+r)/2+1,r,p);
	tr[k].rank=tr[tr[k].l].rank+tr[tr[k].r].rank;
	
}

void del(int k,int l,int r,int p){
	
	if (l==r){
		
		tr[k].rank=0;
		return;
		
	}
	int mid=(l+r)/2;
	if (p<=mid) del(tr[k].l,l,(l+r)/2,p);
	else del(tr[k].r,(l+r)/2+1,r,p);
	tr[k].rank=tr[tr[k].l].rank+tr[tr[k].r].rank;
	
}

int ask_rank(int k,int l,int r,int p){
	
	if (l==r) return l;
	if (p<=tr[tr[k].l].rank) return ask_rank(tr[k].l,l,(l+r)/2,p);
	else return ask_rank(tr[k].r,(l+r)/2+1,r,p-tr[tr[k].l].rank);
	
}

int ask(int k,int l,int r,int o,int p){
	
	if (r<o||l>p) return 0;
	if (o<=l&&r<=p) return tr[k].rank;
	return ask(tr[k].l,l,(l+r)/2,o,p)+ask(tr[k].r,(l+r)/2+1,r,o,p);
	
}

int main(){
	
	int n,T;
	n=read();T=read();
	cnt=1;
	stree(1,1,n);
	while (T--){
		
		int k=read();
		if (k==1){
			
			int l=read();
			l++;
			a[l]=1;
			add(1,1,n,l);
			
		}
		else if (k==2){
			
			int l=read();
			l++;
			a[l]=0;
			del(1,1,n,l);
			
		}
		else if (k==3){
			
			if (tr[1].rank==0) printf("-1\n");
			else printf("%d\n",ask_rank(1,1,n,1)-1);
			
		}
		else if (k==4){
			
			if (tr[1].rank==0) printf("-1\n");
			else printf("%d\n",ask_rank(1,1,n,tr[1].rank)-1);
			
		}
		else if (k==5){
			
			int l=read();
			l++;
			int p=ask(1,1,n,1,l);
			if (a[l]==0) p++;
			if (p==1) printf("-1\n");
			else printf("%d\n",ask_rank(1,1,n,p-1)-1);
			
		}
		else if (k==6){
			
			int l=read();
			l++;
			int p=ask(1,1,n,1,l);
			if (p==tr[1].rank) printf("-1\n");
			else printf("%d\n",ask_rank(1,1,n,p+1)-1);
			
		}
		else{
			
			int l=read();
			l++;
			printf("%d\n",a[l]?1:-1);
			
		}
		
	}
	system("pause");
	
}