[NOI 2004]郁闷的出纳员(bzoj 1503)
本文最后更新于:2017年9月8日 早上
题目描述
OIER公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资。如果他心情好,就可能把每位员工的工资加上一个相同的量。反之,如果心情不好,就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。
工资的频繁调整很让员工反感,尤其是集体扣除工资的时候,一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案。
老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第k多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。
好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?
输入格式
第一行有两个非负整数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是一个正整数。
在初始时,可以认为公司里一个员工也没有。
输出格式
输出的行数为F命令的条数加1。
对于每条F命令,你的程序要输出一行,仅包含一个整数,为当前工资第k多的员工所拿的工资数,如果k大于目前员工的数目,则输出-1。
输出文件的最后一行包含一个整数,为离开公司的员工的总数。
样例
样例输入
9 10
I 60
I 70
S 50
F 2
I 30
S 15
A 5
F 1
F 2
样例输出
10
20
-1
2
数据范围与提示
id | 内容 |
---|---|
1 | I命令的条数不超过100000 |
2 | A命令和S命令的总条数不超过100 |
3 | F命令的条数不超过100000 |
4 | 每次工资调整的调整量不超过1000 |
5 | 新员工的工资不超过100000 |
思路
由于对所有员工都要增减工资,若对每个员工都修改,复杂度高达$O(n)$,这时候,我们可以记录一个基准值now,每次对它修改,用一个Treap维护相对工资,加时直接加,减时del掉工资小于min-now的员工。至于查询第k大,也就是查询排名为t[root].sz-k+1的树。
注意
1.insert时应插入x-now,因为之前工资的增减对才插入的值没影响。
Code:
#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;
const int maxn=1e5+10;
struct node
{
int l,r,v,w,rd,sz;
node(){}
node(int l,int r,int v,int w,int rd,int sz):l(l),r(r),v(v),w(w),rd(rd),sz(sz){}
}t[maxn];
int root=0,sz=0,n,mi,now=0,tot=0,lv=0;
void PushUP(int);
void lturn(int&);
void rturn(int&);
int del(int&,int);
void insert(int&,int);
int getsum(int&,int);
int main()
{
int x;
char op[2];
scanf("%d%d",&n,&mi);
while(n--)
{
scanf("%s%d",op,&x);
if(op[0]=='I')
{
if(x>=mi) insert(root,x-now);
}
else if(op[0]=='A') now+=x;
else if(op[0]=='S')
{
now-=x;
lv+=del(root,mi-now);
}
else
{
if(x<=t[root].sz) printf("%d\n",getsum(root,t[root].sz-x+1)+now);
else printf("-1\n");
}
}
printf("%d\n",lv);
return 0;
}
void PushUP(int k)
{
t[k].sz=t[t[k].l].sz+t[t[k].r].sz+t[k].w;
}
void lturn(int &k)
{
int fa=t[k].r;t[k].r=t[fa].l;t[fa].l=k;t[fa].sz=t[k].sz;PushUP(k);k=fa;
}
void rturn(int &k)
{
int fa=t[k].l;t[k].l=t[fa].r;t[fa].r=k;t[fa].sz=t[k].sz;PushUP(k);k=fa;
}
void insert(int &k,int x)
{
if(!k)
{
k=++sz;
t[k]=node(0,0,x,1,rand(),1);
return;
}
t[k].sz++;
if(t[k].v==x) t[k].w++;
else if(t[k].v>x)
{
insert(t[k].l,x);
if(t[t[k].l].rd<t[k].rd) rturn(k);
}
else
{
insert(t[k].r,x);
if(t[t[k].r].rd<t[k].rd) lturn(k);
}
}
int getsum(int &k,int x)
{
if(k==0) return 0;
if(x<=t[t[k].l].sz) return getsum(t[k].l,x);
else if(x>t[t[k].l].sz+t[k].w) return getsum(t[k].r,x-(t[t[k].l].sz+t[k].w));
else return t[k].v;
}
int del(int &k,int x)
{
int tp;
if(k==0) return 0;
if(t[k].v<x) return t[t[k].l].sz+t[k].w+del(k=t[k].r,x);
else return tp=del(t[k].l,x),t[k].sz-=tp,tp;
}