Codevs 1286 郁闷的出纳员

1286 郁闷的出纳员 2004年NOI全国竞赛
时间限制: 1 s
空间限制: 128000 KB
题目等级 : 大师 Master
题解
查看运行结果
题目描述 Description
OIER公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。

工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。

老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第k多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。

好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难吧？

输入描述 Input Description
第一行有两个非负整数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是一个正整数。

在初始时，可以认为公司里一个员工也没有。

输出描述 Output Description
输出文件的行数为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

数据范围及提示 Data Size & Hint
【约定】

l I命令的条数不超过100000

l A命令和S命令的总条数不超过100

l F命令的条数不超过100000

l 每次工资调整的调整量不超过1000

l 新员工的工资不超过100000

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
struct sbt{
    int left,right,size,data;
}tre[200010];
int sum,n,minn,now,key,top,root;
string ch;
void left_rot(int &x ){
    int y=tre[x].right;
    tre[x].right=tre[y].left;
    tre[y].left=x;
    tre[y].size=tre[x].size;
    tre[x].size=tre[tre[x].left].size+
                    tre[tre[x].right].size+1;
    x=y;
}
void right_rot(int &x){
    int y=tre[x].left;
    tre[x].left=tre[y].right;
    tre[y].right=x;
    tre[y].size=tre[x].size;
    tre[x].size=tre[tre[x].left].size+
                    tre[tre[x].right].size+1;
    x=y;
}
void maintain(int &x,bool flag){
    if(!flag){
        if(tre[tre[tre[x].left].left].size>
            tre[tre[x].right].size) right_rot(x);
        else{
            if(tre[tre[tre[x].left].right].size>
                tre[tre[x].right].size)
            left_rot(tre[x].left),right_rot(x);
            else return;
        }
    }
    else {
        if(tre[tre[tre[x].right].right].size>
            tre[tre[x].left].size)
                left_rot(x);
        else {
            if(tre[tre[tre[x].right].left].size>
                tre[tre[x].left].size)
            right_rot(tre[x].right),left_rot(x);
            else return ;
        }
    }
    maintain(tre[x].left,0); maintain(tre[x].right,1);
    maintain(x,1); maintain(x,0);
}
void insert(int &x,int data){
    if(x==0){
        x= ++top;
        tre[x].left=tre[x].right=0;
        tre[x].size=1;
        tre[x].data=data;
    }
    else {
        tre[x].size++;
        if(data<tre[x].data) insert(tre[x].left,data);
        else insert(tre[x].right,data);
        maintain(x,data >= tre[x].data);
    }
}
int remove(int &x,int data){
    tre[x].size--;
    if(tre[x].data>data) remove(tre[x].left,data);
    else if(tre[x].data<data)
        remove(tre[x].right,data);
    else{
        if(tre[x].left!=0&&tre[x].right==0){
            int ret=x;
            x=tre[x].left;
            return ret;
        }
        else if(tre[x].left==0&&tre[x].right!=0){
            int ret=x;
            x=tre[x].right;
            return ret;
        }
        else if(tre[x].left==0&&tre[x].right==0){
            int ret=x;
            x=0; return ret;
        }
        else if(tre[x].left!=0&&tre[x].right!=0){
            int ret=tre[x].right;
            while(tre[ret].left) ret=tre[ret].left;
            tre[x].data=tre[ret].data;
            remove(tre[x].right,tre[ret].data);
        }
    }
}
int select(int &x,int k){
    int r=tre[tre[x].left].size+1;
    if(r==k) return tre[x].data;
    else if(r<k) return select(tre[x].right,k-r);
    else if(r>k) return select(tre[x].left,k);
}
int getmin(){
    int x;
    for(x=root;tre[x].left;x=tre[x].left);
    return tre[x].data;
}
int main(){
    scanf("%d%d",&n,&minn);
    for(int i=1;i<=n;i++){
        cin>>ch;
        scanf("%d",&key);
        if(now){
            while(getmin()<minn){
                remove(root,getmin());
                sum++;now--;
            }
        }
        if(ch=="I"){
            if(key>=minn) insert(root,key),now++;
        }
        else if (ch=="A")
            for(int j=1;j<=top;j++)
                tre[j].data+=key;
        else if (ch=="S")
            for(int j=1;j<=top;j++)
                tre[j].data-=key;
        else if(ch=="F"){
            if(key>now){
                printf("-1\n");
                continue;
            }
            else 
            printf("%d\n",select(root,now-key+1));
        }
    }
    cout<<sum;
    return 0;
}

感觉自己好久没来写代码了，键盘都陌生了，数学、背包……耽误好多时间、、、怎么办呐！！！