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

猜你喜欢

转载自www.cnblogs.com/chhokmah/p/10578146.html