BZOJ 1503 [NOI 2004] 郁闷的出纳员 （splay）

题目链接：BZOJ 1503

每次写数据结构，都有一种debug到心力交瘁的感觉，我比出纳员还郁闷啊= =。感觉自己写这道题被各种东西坑。它是求第k大的工资，我开始求的是第k小的工资。对于那些一开始的工资就小于标准线的，不计入离开公司的人数里。还有插入和删除各种恶心，每个都要递归处理。我开始写双旋root更新不了，最后只有改。好伤心啊，看来这几天我都得刷数据结构的题了QAQ。

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define maxn (100000+10)
int N,M=0,K,root=0,tot=0;
int c[maxn][2],fa[maxn],size[maxn],v[maxn];

void pushup(int x){
	int l=c[x][0],r=c[x][1];
	size[x]=size[l]+size[r]+1;
}

void rotate(int &x){
	int y=fa[x],z=fa[y],l,r;
	if(c[y][0]==x)l=0; else l=1; r=l^1;
	if(c[z][0]==y)c[z][0]=x;
	else c[z][1]=x;
	fa[c[x][r]]=y; fa[y]=x; fa[x]=z;
	c[y][l]=c[x][r]; c[x][r]=y;
	pushup(y); pushup(x);
	if(y==root)root=x;//因为此时x为y的父亲节点，若y为root的话，就应该更新root 
}

void splay(int &x,int k){
	while(fa[x]!=k){//妈蛋，直接双旋总会挂，我只有这么写了= =
		rotate(x);
	}
}

int find(int x,int rk){//因为是求第k大，所以与x的右儿子比较
	int l=c[x][0],r=c[x][1];
	if(size[r]+1==rk)return v[x];
	else if(size[r]>=rk)return find(r,rk);
	else return find(l,rk-size[r]-1);
}

void ins(int k){//针对那种一开始没有建好图，边读入，边插入的情况
	if(!root){
		root=++tot;
		v[tot]=k; size[tot]=1; fa[tot]=0;
		return ;
	}
	int t=root,f;
	while(t){//t即为要插入值为k的节点的父亲节点
		f=t; ++size[t];
		if(k<v[t])t=c[t][0]; else t=c[t][1];
	}
	if(v[f]>k)c[f][0]=++tot; else c[f][1]=++tot;
	v[tot]=k; size[tot]=1; fa[tot]=f;
	splay(tot,0);//将该节点旋转到0的儿子节点
}

int dec(int f,int &x){//递归处理
	if(!x)return 0;
	int temp;
	if(v[x]+M<K){
		temp=dec(x,c[x][1])+size[c[x][0]]+1;
		size[c[x][1]]=size[x]-temp;
		x=c[x][1]; fa[x]=f;
	}
	else{
		temp=dec(x,c[x][0]);
		size[x]-=temp;
	}
	return temp;
}

int main(){
	scanf("%d%d",&N,&K);
	char s[10]; int x,ans=0;
	for(int i=1;i<=N;i++){
		scanf("%s%d",s,&x);
		if(s[0]=='I'&&x>=K)ins(x-M);
		if(s[0]=='A')M+=x;
		if(s[0]=='S'){M-=x;ans+=dec(0,root);}
		if(s[0]=='F'){
			if(size[root]<x)printf("-1\n");
			else printf("%d\n",find(root,x)+M);
		}
	}
	printf("%d\n",ans);
	return 0;
}