NOI2004——郁闷的出纳员（SBT）

郁闷的出纳员（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
;

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;
}