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*2struct 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;}