luogu P1486 [NOI2004]郁闷的出纳员 |Splay

题目描述

OIER 公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把当前在公司的所有员工的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。

工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。

老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第 kkk 多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。

好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难吧？

如果某个员工的初始工资低于最低工资标准，那么将不计入最后的答案内。

输入格式

第一行有两个整数 \(n\) 和 \(⁡\min\)。\(n\) 表示下面有多少条命令，\(\min\) 表示工资下界。

接下来的 \(n\) 行，每行一个字符 \(x\) 和一个整数 \(k\)，表示一条命令。命令可以是以下四种之一：

I k 新建一个工资档案，初始工资为 k。如果某员工的初始工资低于工资下界，他将立刻离开公司。

A k 把每位员工的工资加上 k 。

S k 把每位员工的工资扣除 k。

F k 查询第 k 多的工资。

在初始时，可以认为公司里一个员工也没有。

输出格式

对于每条 F 命令，你的程序要输出一行，仅包含一个整数，为当前工资第 \(k\) 多的员工所拿的工资数，如果 \(k\) 大于目前员工的数目，则输出 \(-1\)。

输出文件的最后一行包含一个整数，为离开公司的员工的总数。

---

#include <bits/stdc++.h>
using namespace std;
const int N=1e6+5,INF=1<<30;
inline int read(){
	int x=0,f=1; char c=getchar();
	while(c<'0'||c>'9'){ if(c=='-')f=-1;  c=getchar(); }
	while('0'<=c&&c<='9'){ x=(x<<1)+(x<<3)+(c^48); c=getchar();  }
	return x*f;	
}
int ch[N][2],par[N],val[N],cnt[N],size[N],ncnt,root;
inline bool chk(int x){ return ch[par[x]][1]==x; }
inline void pushup(int x){ size[x]=size[ch[x][0]]+size[ch[x][1]]+cnt[x]; }
inline void rotate(int x){
	int y=par[x],z=par[y],k=chk(x),w=ch[x][k^1];
	ch[z][chk(y)]=x,par[x]=z;
	ch[y][k]=w,par[w]=y;
	ch[x][k^1]=y,par[y]=x;
	pushup(y),pushup(x);
}
inline void splay(int x,int goal=0){
	while(par[x]!=goal){
		int y=par[x],z=par[y];
		if(z!=goal)chk(x)==chk(y)?rotate(y):rotate(x);	
		rotate(x);
	}
	if(!goal)root=x;
}
inline void insert(int x){
	int p=root,cur=0;
	while(p&&val[p]!=x)cur=p,p=ch[p][x>val[p]];
	if(p)cnt[p]++;
	else {
		p=++ncnt;
		if(cur)ch[cur][x>val[cur]]=p;
		ch[p][0]=ch[p][1]=0;
		par[p]=cur,val[p]=x;
		cnt[p]=size[p]=1;
	}
	splay(p);
}
inline void find(int x){
	int p=root;
	while(ch[p][x>val[p]]&&x!=val[p])p=ch[p][x>val[p]];
	splay(p);
}
inline int pre(int x){
	find(x);
	if(val[root]<x)return root;
	int p=ch[root][0];
	while(ch[p][1])p=ch[p][1];
	return p;	
}
inline int succ(int x){
    find(x);
    if(val[root]>x)return root;
    int p=ch[root][1];
    while(ch[p][0])p=ch[p][0];
    return p;
}
inline void remove(int x){
	int last=pre(x),nxt=succ(x);
	splay(last),splay(nxt,last);
	int p=ch[nxt][0];
	if(cnt[p]>1)cnt[p]--,splay(p);
	else ch[nxt][0]=0;	
}
inline int kth(int k){
	int p=root;
	while(1){
		if(ch[p][0]&&k<=size[ch[p][0]])p=ch[p][0];
		else if(k>size[ch[p][0]]+cnt[p]){
			k-=size[ch[p][0]]+cnt[p];
			p=ch[p][1];
		}else return p;
	}
}
int n,Min;
signed main(){
	n=read(),Min=read();
	int ans=0,delta=0,totadd=0;
	insert(INF);
	insert(-INF);
	
	char c[3];
	while(n--){
		scanf("%s",c);
		int k=read();
		if(c[0]=='I'){
			if(k<Min)continue;
			insert(k-delta);	
			totadd++;
		}else if(c[0]=='A'){
			delta+=k;
		}else if(c[0]=='S'){
			delta-=k;
			insert(Min-delta);
			find(-INF);
			int a=root;
			find(Min-delta);
			int b=root;
			splay(a),splay(b,a);
			ch[ch[root][1]][0]=0;
			remove(Min-delta);
		}else if(c[0]=='F'){
			find(INF);
			int now=size[ch[root][0]]-1;
			if(now<k){ printf("-1\n"); continue; }
			int ans=val[kth(now+2-k)]+delta;
			printf("%d\n",ans);
		}
	}
	find(INF);
	cout<<totadd-size[ch[root][0]]+1<<endl;
}