输入描述 Input Description
第一行有两个非负整数n和min。n表示下面有多少条命令,min表示工资下界。
接下来的n行,每行表示一条命令。命令可以是以下四种之一:
名称 格式 作用
I命令 I_k 新建一个工资档案,初始工资为k。如果某员工的初始工资低于工资下界,他将立刻离开公司。
A命令 A_k 把每位员工的工资加上k
S命令 S_k 把每位员工的工资扣除k
F命令 F_k 查询第k多的工资
_(下划线)表示一个空格,I命令、A命令、S命令中的k是一个非负整数,F命令中的k是一个正整数。
在初始时,可以认为公司里一个员工也没有。
输出描述 Output Description
输出文件的行数为F命令的条数加一。
对于每条F命令,你的程序要输出一行,仅包含一个整数,为当前工资第k多的员工所拿的工资数,如果k大于目前员工的数目,则输出-1。
输出文件的最后一行包含一个整数,为离开公司的员工的总数。
样例输入 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
经典splay。参考黄学长代码
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int sum[100001],t[100001],fa[100001];
int ts[100001][2],m,size=0,root,delta,l;
int read(){
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void rotate(int x,int &k){
int y=fa[x],z=fa[y],l,r;
if (ts[y][0]==x)l=0;else l=1;r=l^1;
if (y==k) k=x;
else if (ts[z][0]==y)ts[z][0]=x;else ts[z][1]=x;
fa[x]=z;fa[y]=x;
ts[y][l]=ts[x][r];
fa[ts[x][r]]=y;
ts[x][r]=y;
sum[x]+=1+sum[ts[y][r]];
sum[y]=sum[y]-1-sum[ts[x][l]];
}
void splay(int x,int &k){
int y,z;
while (x!=k){
y=fa[x];z=fa[y];
if (y!=k){
if (ts[z][0]==y^ts[y][0]==x)rotate(x,k);else rotate(y,k);
}
rotate(x,k);
}
}
void ins(int &k,int x,int pre){
if (k==0){
size++;k=size;
t[k]=x;sum[k]=1;fa[k]=pre;
splay(k,root);
return;
}
sum[k]++;
if (x>=t[k]) ins(ts[k][1],x,k);else ins(ts[k][0],x,k);
}
int find(int k,int x){
if (x==sum[ts[k][1]]+1) return t[k];
if (sum[ts[k][1]]>=x) return find(ts[k][1],x);
else return find(ts[k][0],x-1-sum[ts[k][1]]);
}
int dec(int &x,int pre){
if(x==0)return 0;
int k;
if(t[x]+delta<m){
k=dec(ts[x][1],x)+sum[ts[x][0]]+1;
sum[ts[x][1]]=sum[x]-k;
x=ts[x][1];fa[x]=pre;
}
else{
k=dec(ts[x][0],x);
sum[x]-=k;
}
return k;
}
int main(){
int n=read();m=read();
for (int i=1;i<=n;i++){
char s[2];
scanf("%s",s);
int k=read();
if (s[0]=='I'&&k>=m)ins(root,k-delta,0);
if (s[0]=='A')delta+=k;
if (s[0]=='S'){
delta-=k;
l+=dec(root,0);
}
if (s[0]=='F'){
printf("%d\n",k<=sum[root]?find(root,k)+delta:-1);
}
}
printf("%d",l);
return 0;
}