BZOJ_P1503&Codevs_P1286 [NOI2004]郁闷的出纳员(SplayTree)

BZOJ传送门

Codevs传送门

Time Limit: 5 Sec Memory Limit: 64 MB

Submit: 9079 Solved: 3161

[Submit][Status][Discuss]

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

Source

Splay

根据数据减法不超过100，直接在线递归减即可不用标记，先遍历，根据题目可以进行特殊删点，因为它左节点肯定是null！，直接将右节点接上即可(内存也比较小不担心内存问题)

void remove(Node* &o,int x){
if(o==null) return;
if(o->v==x)
o=o->ch[1];
o->maintain();
}

#include<cstdio>
#include<cstring>
#include<climits>
#include<cstdlib>
#include<iostream>
using namespace std;
#define INF INT_MAX/3*2
struct SplayTree{
struct Node{
Node *ch[2];
int r,v,s,size;
Node(int v,Node* nl):v(v){size=s=1;r=rand();ch[0]=ch[1]=nl;}
void maintain(){size=ch[0]->size+s+ch[1]->size;}
}*root,*null;
int minv,tot,leave;
SplayTree(){
null=new Node(0,0);null->r=INT_MAX;null->size=null->s=0;
null->ch[0]=null->ch[1]=null;root=null;
}
void rotate(Node* &o,int d) {
Node* k=o->ch[d^1];o->ch[d^1]=k->ch[d],k->ch[d]=o;
o->maintain(),k->maintain();o=k;
}
void insert(Node* &o,int x){
if(o==null){o=new Node(x,null);return;}
if(o->v==x){o->s++,o->size++;return;}
insert(o->ch[x>o->v],x);
if(o->ch[x>o->v]->r<o->r) rotate(o,x<o->v);
else o->maintain();
}
void remove(Node* &o,int x){
if(o==null) return;
if(o->v==x)
o=o->ch[1];
o->maintain();
}
int kth(Node *o,int x){
if(o==null) return -1;
int d=o->ch[1]->size;
if(x<=d) return kth(o->ch[1],x);
else if(x>d+o->s) return kth(o->ch[0],x-d-o->s);
return o->v;
}
void add(Node* &o,int k){
if(o==null) return;
o->v+=k;add(o->ch[0],k);add(o->ch[1],k);
}
void sub(Node* &o,int k){
if(o==null) return;
o->v-=k;sub(o->ch[0],k);sub(o->ch[1],k);
if(o->v<minv){
leave+=o->s;
remove(o,o->v);
}
}
}T;
int in(){
int x=0,w;char ch=getchar();
while(ch==' '||ch=='\n') ch=getchar();
if(ch=='+') w=1;if(ch=='-') w=0;ch=getchar();
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*w;
}
void out(int x){
char ch[32];int l=0;memset(ch,0,sizeof(ch));
while(x){
ch[l++]=x%10;x/=10;
}
for(int i=l-1;i>=0;i--) putchar(ch[i]+'0');
}
int main(){
int n,k;char ch;
scanf("%d%d",&n,&T.minv);ch=getchar();
while(n--){
while(ch>'Z'||ch<'A') ch=getchar();
scanf("%d",&k);
switch(ch){
case 'I':
if(k>=T.minv) T.insert(T.root,k);
break;
case 'S':
T.sub(T.root,k);
break;
case 'A':
T.add(T.root,k);
break;
case 'F':
if(T.root==T.null||T.root->size<k) printf("%d\n",-1);
else printf("%d\n",T.kth(T.root,k));
break;
}
ch=getchar();
}
printf("%d",T.leave);
return 0;
}