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