[bzoj1503][NOI2004]郁闷的出纳员

Description

\(OIER\)公司是一家大型专业化软件公司,有着数以万计的员工.
作为一名出纳员,我的任务之一便是统计每位员工的工资.
这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资:如果他心情好,就可能把每位员工的工资加上一个相同的量;反之,如果心情不好,就可能把他们的工资扣除一个相同的量.我真不知道除了调工资他还做什么其它事情.
工资的频繁调整很让员工反感,尤其是集体扣除工资的时候,一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了.每位员工的工资下界都是统一规定的.每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案.
老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第\(k\)多的员工拿多少工资.每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案.
好了,现在你已经对我的工作了解不少了.正如你猜的那样,我想请你编一个工资统计程序.怎么样,不是很困难吧?

Input

第一行两个非负整数\(n,min.n\)表示下面有多少条命令,\(min\)表示工资下界.
接下来的\(n\)行,每行表示一条命令,命令可以是以下四种之一:

1. \(I\;k\):新建一个工资档案,初始工资为\(k\).如果某员工的初始工资低于工资下界,他将立刻离开;
2. \(A\;k\):把每位员工的工资\(+k\);
3. \(S\;k\):把每位员工的工资\(-k\);
4. \(F\;k\):查询第\(k\)多的工资.
   \(I,A,S\)命令中的\(k\in{N}\),\(F\)命令中的\(k\in{N^{+}}\).
   初始时,可以认为公司里一个员工也没有(感觉好迷啊).

Output

输出文件的行数为\(F\)命令的条数\(+1\).
对于每条\(F\)命令,你的程序要输出一行,仅包含一个整数,为当前工资第\(k\)多的员工所拿的工资数,如果k大于目前员工的数目,则输出\(-1\).
输出文件的最后一行包含一个整数,为离开公司的员工的总数(\(I\)中立刻离开的不算).

Sample Input

9 10
I 60
I 70
S 50
F 2
I 30
S 15
A 5
F 1
F 2

Sample Output

10
20
-1
2

HINT

\(I\)命令的条数\(\leq100000,A,S\)命令的总条数\(\leq100,F\)命令的条数\(\leq100000\),每次工资调整的调整量\(\leq1000\),新员工的工资\(\leq100000\).

Solution

记录总变化量,\(splay\)直接上.

#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<cstdio>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define N 100005
using namespace std;
struct BBT{
	int c[2],f,siz,cnt,val;
}tr[N];
int n,m,cnt,tot,sum,root;
inline int read(){
	int ret=0;char c=getchar();
	while(!isdigit(c))
		c=getchar();
	while(isdigit(c)){
		ret=(ret<<1)+(ret<<3)+c-'0';
		c=getchar();
	}
	return ret;
}
inline int find(int x){
	int u=root;
	while(x!=tr[u].val&&u){
		if(x<tr[u].val)
			u=tr[u].c[0];
		else u=tr[u].c[1];
	}
	return u;
}
inline bool son(int u){
	return tr[tr[u].f].c[1]==u;
}
inline int recnt(int u){
	tr[u].siz=tr[tr[u].c[0]].siz+tr[tr[u].c[1]].siz+tr[u].cnt;
}
inline void ins_p(int f,int u,int c){
	tr[f].c[c]=u;tr[u].f=f;
}
inline void rotate(int u){
	int f=tr[u].f;bool c=son(u);
	ins_p(tr[f].f,u,son(f));
	ins_p(f,tr[u].c[c^1],c);
	ins_p(u,f,c^1);
	recnt(f);recnt(u);
	if(!tr[u].f) root=u;
}
inline void splay(int u){
	while(tr[u].f){
		if(!tr[tr[u].f].f) rotate(u);
		else{
			if(son(u)==son(tr[u].f)){
				rotate(tr[u].f);rotate(u); 
			}
			else{
				rotate(u);rotate(u);
			}
		}
	} 
}
inline void insert(int x){
	if(x+sum<m){
		return;
	}
	int u=find(x),f;
	if(u){
		++tr[u].cnt;
		while(u){
			recnt(u);u=tr[u].f;
		}
		return;
	}
	u=root;f=0;
	while(u){
		f=u;
		if(x<tr[u].val)
			u=tr[u].c[0];
		else u=tr[u].c[1];
	}
	tr[++cnt].val=x;
	tr[cnt].cnt=tr[cnt].siz=1;
	if(!f){
		root=cnt;
		recnt(root);
		return;
	}
	if(x<tr[f].val) ins_p(f,cnt,0);
	else ins_p(f,cnt,1);
	splay(cnt);
}
inline int near(int u,int c){
	if(tr[u].c[c]){
		u=tr[u].c[c];
		while(tr[u].c[c^1])
			u=tr[u].c[c^1];
		return u;
	}
	if(son(u)==c^1) return tr[u].f;
	u=tr[u].f;
	while(son(u)==c){
		if(u==root) return u;
		u=tr[u].f;
	}
	return tr[u].f;
}
inline void clear(int u){
	tr[tr[u].f].c[son(u)]=0;
	tr[u].val=0;
	tr[u].siz=tr[u].cnt=0;
	recnt(tr[u].f);
	tr[u].f=0;
	tr[u].c[0]=tr[u].c[1]=0;
}
inline int del(int u,int x){
	if(!u) return 0;
	int ret;
	if(tr[u].val<x){
		ret=tr[tr[u].c[0]].siz+tr[u].cnt+del(tr[u].c[1],x);
		tr[tr[u].c[1]].f=tr[u].f;
		tr[tr[u].f].c[son(u)]=tr[u].c[1];
		tr[tr[u].c[1]].siz=tr[u].siz-ret;
		if(!tr[u].f) root=tr[u].c[1];
	}
	else{
		ret=del(tr[u].c[0],x);
		tr[u].siz-=ret;
	}
	return ret;
}
inline int ask(int k){
	int u=root;
	if(k>tr[u].siz) return -1-sum;
	while(true){
		if(tr[u].c[1]&&k<=tr[tr[u].c[1]].siz)
			u=tr[u].c[1];
		else{
			k-=tr[tr[u].c[1]].siz;
			if(k<=tr[u].cnt)
				return tr[u].val;
			k-=tr[u].cnt;
			u=tr[u].c[0];
		}
	}
}
inline void print(int u){
	if(!u) return;
	printf("%d->(%d,%d),",u,tr[u].c[0],tr[u].c[1]);
	printf("cnt=%d,siz=%d,val=%d,f=%d\n"\
	  ,tr[u].cnt,tr[u].siz,tr[u].val+sum,tr[u].f); 
	print(tr[u].c[0]);print(tr[u].c[1]);
}
inline void Aireen(){
	n=read();m=read();char c[2];
	while(n--){
		scanf("%s",c);
		if(c[0]=='I') insert(read()-sum);
		else if(c[0]=='A') sum+=read();
		else if(c[0]=='S'){
			sum-=read();
			tot+=del(root,m-sum);//k+sum<m => k<m-sum
		}
		else if(c[0]=='F')
			printf("%d\n",ask(read())+sum);
	}
	printf("%d\n",tot);
}
int main(){
	freopen("cashier.in","r",stdin);
	freopen("cashier.out","w",stdout);
	Aireen();
	fclose(stdin);
	fclose(stdout);
	return 0;
}

2017-03-28 23:39:39