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