郁闷的出纳员(cashier.pas)
【问题描述】
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命令的条数加一。
对于每条F命令,你的程序要输出一行,仅包含一个整数,为当前工资第k多的员工所拿的工资数,如果k大于目前员工的数目,则输出-1。
输出文件的最后一行包含一个整数,为离开公司的员工的总数。
【样例输入】(cashier.in)
9 10
I 60
I 70
S 50
F 2
I 30
S 15
A 5
F 1
F 2
【样例输出】(cashier.out)
10
20
-1
2
【约定】
l I命令的条数不超过100000
l A命令和S命令的总条数不超过100
l F命令的条数不超过100000
l 每次工资调整的调整量不超过1000
l 新员工的工资不超过100000
【评分方法】
对于每个测试点,如果你输出文件的行数不正确,或者输出文件中含有非法字符,得分为0。
否则你的得分按如下方法计算:如果对于所有的F命令,你都输出了正确的答案,并且最后输出的离开公司的人数也是正确的,你将得到10分;如果你只对所有的F命令输出了正确答案,得6分;如果只有离开公司的人数是正确的,得4分;否则得0分。
解析:
这是一道数据结构题,据说splay,AVL,SBT,Treap,线段树 ,树状数组等都可以实现
这两天刚学了SBT,虽然部分代码理解的还不是很透彻,但大致思想还是掌握了。。。
这里推荐两个学习SBT的地址:
讲解:http://www.nocow.cn/index.php/Size_Balanced_Tree
代码(包括SBT的九种操作):http://www.docin.com/p-481614045.html
代码:
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=100010;
int root,tot;
struct SBT
{
int left,right,size,key;
void init()
{
left=right=key=0;
size=1;
}
}tree[N];
void go_left(int &p)
{
int k=tree[p].right;
tree[p].right=tree[k].left;
tree[k].left=p;
tree[k].size=tree[p].size;
tree[p].size=tree[tree[p].left].size+tree[tree[p].right].size+1;
p=k;
}
void go_right(int &p)
{
int k=tree[p].left;
tree[p].left=tree[k].right;
tree[k].right=p;
tree[k].size=tree[p].size;
tree[p].size=tree[tree[p].left].size+tree[tree[p].right].size+1;
p=k;
}
void maintain(int &p,bool flag)
{
if(flag)
{
if(tree[tree[tree[p].right].right].size>tree[tree[p].left].size)go_left(p);
else if(tree[tree[tree[p].right].left].size>tree[tree[p].left].size)
{
go_right(tree[p].right);
go_left(p);
}
else return ;
}
else //否则更新左子树
{
if(tree[tree[tree[p].left].left].size>tree[tree[p].right].size)go_right(p);
else
if(tree[tree[tree[p].left].right].size>tree[tree[p].right].size)
{
go_left(tree[p].left);
go_right(p);
}
else return ;
}
maintain(tree[p].left,false);
maintain(tree[p].right,true);
maintain(p,false);
maintain(p,true);
}
void insert(int &p,int k)
{
if(p==0)
{
p=++tot;
tree[p].init();
tree[p].key=k;
}
else
{
tree[p].size++;
if(k<tree[p].key)
insert(tree[p].left,k);
else
insert(tree[p].right,k);
maintain(p,k>=tree[p].key);
}
}
void Delete(int &p,int delay,int min_val)
{
if(!p)return ;
if(tree[p].key+delay<min_val)
{
p=tree[p].right;
Delete(p,delay,min_val);
}
else
{
Delete(tree[p].left,delay,min_val);
tree[p].size=tree[tree[p].left].size+tree[tree[p].right].size+1;
}
}
int get_max_kth(int &p,int k)
{
int t=tree[tree[p].right].size+1;
if(t==k)return tree[p].key;
if(t<k)return get_max_kth(tree[p].left,k-t);
else return get_max_kth(tree[p].right,k);
}
void readdata()
{
freopen("cashier.in","r",stdin);
freopen("cashier.out","w",stdout);
}
void work()
{
int n,min_val,delay;
while(scanf("%d%d",&n,&min_val)!=EOF)
{
tot=delay=root=0;
while(n--)
{
char op[10];
int k;
scanf("%s%d",&op,&k);
if(op[0]=='I')
{
if(k<min_val)continue;
insert(root,k-delay);
}
else if(op[0]=='A')delay+=k;
else if(op[0]=='F')printf("%d\n",tree[root].size<k?-1:(get_max_kth(root,k)+delay));
else
{
delay-=k;
Delete(root,delay,min_val);
}
}
printf("%d\n",tot-tree[root].size);
}
}
int main()
{
readdata();
work();
return 0;
}