Description
OIER公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资。如果他心情好,就可能把每位员工的工资加上一个相同的量。反之,如果心情不好,就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。工资的频繁调整很让员工反感,尤其是集体扣除工资的时候,一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案。老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第k多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?
Input
Output
输出文件的行数为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
HINT
I命令的条数不超过100000 A命令和S命令的总条数不超过100 F命令的条数不超过100000 每次工资调整的调整量不超过1000 新员工的工资不超过100000
分析:
普通的splay…
不过要加一个变量w ,然后每次A操作时w+=num.
S操作:w-=num,且把minn-w(最低工资)在树中的前驱旋到根之后把离开公司总人数sum加上该点元素个数+左子树元素个数,同时更新root.
I操作:如果员工进入时工资x就低于min则直接踢出,且不算入离开公司总人数;
否则把x-w加入树中.
这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int N=200001;
int ch[N][2],pre[N],size[N],cnt[N],v[N];
int top,root,tot=0,w,lim,ans,n,minn; //w记录加减情况
void clear(int x)
{
ch[x][0]=ch[x][1]=pre[x]=size[x]=cnt[x]=v[x]=0;
}
bool get(int x)
{
return ch[pre[x]][1]==x; //x是不是爸爸的右儿子
}
void update(int x)
{
size[x]=cnt[x];
if (ch[x][0]) size[x]+=size[ch[x][0]];
if (ch[x][1]) size[x]+=size[ch[x][1]];
}
void rotate(int x)
{
int fa=pre[x];
int grand=pre[fa];
int wh=get(x);
ch[fa][wh]=ch[x][wh^1];
pre[ch[fa][wh]]=fa;
ch[x][wh^1]=fa;
pre[fa]=x;
pre[x]=grand;
if (grand)
{
ch[grand][ch[grand][1]==fa ? 1:0]=x;
}
update(fa); update(x); //////
return;
}
void splay(int x)
{
for (int fa;fa=pre[x];rotate(x))
if (pre[fa])
rotate(get(x)==get(fa)? fa:x);
root=x;
}
void insert(int x)
{
if (root==0)
{
top++;
cnt[top]=1; ch[top][0]=ch[top][1]=pre[top]=0;
size[top]=1; cnt[top]=1; v[top]=x;
root=top;
return;
}
int now=root,fa=0;
while (1)
{
if (v[now]==x)
{
cnt[now]++;
update(now);
update(fa);
splay(now);
return;
}
fa=now;
now=ch[now][x>v[now]];
if (now==0)
{
top++;
ch[top][0]=ch[top][1]=0;
pre[top]=fa;
size[top]=1;
cnt[top]=1; v[top]=x;
ch[fa][x>v[fa]]=top; //
update(fa);
splay(top);
return;
}
}
}
int find(int x)
{
int now=root;
while (1)
{
if (ch[now][0]&&x<=size[ch[now][0]]) now=ch[now][0];
else{
int temp=(ch[now][0]?size[ch[now][0]]:0)+cnt[now];
if (x<=temp)
return v[now];
x-=temp; now=ch[now][1];
}
}
}
void fro(int k,int x) //找前驱
{
if (!k) return;
if (v[k]<x) {
ans=k;
fro(ch[k][1],x);
}
else fro(ch[k][0],x);
}
int main()
{
scanf("%d%d",&n,&minn);
for (int i=1;i<=n;i++)
{
char cha,c;
int opt;
scanf("%c",&c);
scanf("%c%d",&cha,&opt);
if (cha=='I')
{
if (opt>=minn) insert(opt-w); //差值
}
else if (cha=='A')
{
w+=opt;
}
else if (cha=='S')
{
w-=opt;
lim=minn-w; //把minn-w的在树中的前驱旋到根
ans=1000000;
fro(root,lim); //前驱
if (ans!=1000000)
{
splay(ans);
tot+=size[ch[root][0]]+cnt[root];
root=ch[root][1];
pre[root]=0;
update(root);
}
}
else {
int sz=size[root];
if (opt>sz) printf("-1
");
else printf("%d
",find(sz-opt+1)+w);
//找第k大->找排名sz-k+1的值
}
}
printf("%d
",tot);
return 0;
}