[luogu1486][bzoj1503][NOI2004]郁闷的出纳员【平衡树treap】

题目描述

OIER公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。
工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。
老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第k多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。
好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难吧？
如果某个员工的初始工资低于最低工资标准,那么将不计入最后的答案。

分析

模拟赛的题目是这道题目的改编，成功把自己卡成40分QwQ。
其实就是一个treap的模板，在下降的时候要把整体一起下降。
I命令：加入一个员工 我们在平衡树中加入k-minn
A命令：把每位员工的工资加上k delta加k即可
S命令：把每位员工的工资扣除k 此时我们就需要考虑会不会导致一大批员工离开；我们插入minn-delta，然后使小于minn-delta的点一起移动到根的右子树的左子树，一举消灭；
最后一个命令就是直接查询kth就可以了。
代码好久之前码的，码风不大一样。

代码

# include <bits/stdc++.h>
# define N 100005
using namespace std ;
struct Treap{
    int ch[N][2] , rd[N] , sz[N] , v[N] , num[N] ;
    int root , sum , delta ;
    void maintain( int &o ) { sz[o] = sz[ch[o][0]] + sz[ch[o][1]] + 1 ; }
    void right_rotate( int &o ) {
        int lson = ch[o][0] ; ch[o][0] = ch[lson][1] ; ch[lson][1] = o ; sz[lson] = sz[o] ;
        maintain(o) ; o = lson ;
    }
    void left_rotate( int &o ) {
        int rson = ch[o][1] ; ch[o][1] = ch[rson][0] ; ch[rson][0] = o ; sz[rson] = sz[o] ;
        maintain(o) ; o = rson ;
    }
    void insert( int &o , int x ) {
        if ( o == 0 ) { sum ++ ; o = sum ; rd[o] = rand() ; num[o] = x ; sz[o] = 1 ; return ; }
        sz[o] ++ ;
        if ( x < num[o] ) { insert( ch[o][0] , x ) ; if ( rd[ch[o][0]] < rd[o] ) right_rotate(o) ; }
        else { insert( ch[o][1] , x ) ; if ( rd[ch[o][1]] < rd[o] ) left_rotate(o) ; }
    }
    int del( int &o , int x ) {
        int tmp ; if ( o == 0 ) return 0 ;
        if ( num[o] < x ) { tmp = sz[ch[o][0]] + 1 ; o = ch[o][1] ; return tmp + del( o , x ) ; }
        else { tmp = del( ch[o][0] , x ) ; sz[o] -= tmp ; return tmp ; }
    }
    int find( int o , int x ) {
        if ( sz[ch[o][0]] + 1 == x ) return num[o] + delta ;
        else if ( sz[ch[o][0]] + 1 < x ) return find( ch[o][1] , x - sz[ch[o][0]] - 1 ) ;
        else return find( ch[o][0] , x ) ;
    }
};
Treap Tp ;
int n , m , cnt ;
int main () {
    scanf( "%d%d" , &n , &m ) ;
    for ( int i = 1 ; i <= n ; i ++ ) {
        char s[10] ; int x ;
        scanf( "%s%d" , s , &x ) ;
        if ( s[0] == 'I' ) if ( x >= m ) Tp.insert( Tp.root , x - Tp.delta ) ;
        if ( s[0] == 'A' ) Tp.delta += x ;
        if ( s[0] == 'S' ) Tp.delta -= x , cnt += Tp.del( Tp.root , m - Tp.delta ) ;
        if ( s[0] == 'F' ) {
            if ( x > Tp.sz[Tp.root] ) printf( "-1\n" ) ;
            else printf( "%d\n" , Tp.find( Tp.root , Tp.sz[Tp.root] - x + 1 ) ) ;
        }
        //cout << "cnt : " << cnt << endl ;
    }
    printf( "%d\n" , cnt ) ;
    return 0 ;
}