BZOJ 1503 [NOI2004] 郁闷的出纳员 treap

题意: 链接

方法: treap

解析: 这是本蒟蒻的第二道treap题,第二遍写的时候update,左旋右旋,插入函数都可以大概写出来了(还是得练啊),然而del  函数却被虐了,自己也想到用个什么基准来记(不过怎么写基准啊喂!),还是看了题解才看明白基准的用法和del函数,  这题不像做的第一道,所以tr[k].w这个变量没必要存在,加了会超时别问我为什么.总之,还是得练啊

附上讲解链接 : http://blog.csdn.net/u014141559/article/details/43057817

为了方便考虑,定义基准值为delta,相对工资V对应的实际工资为F[V],则有F[V]=V+deltaV=F[V]-delta。定义工资下限为lowbound这是一个实际的下限,存储相对下限就是lowbound-delta

对于I_k插入一个新的工资记录值kk为实际工资,对应的相对工资为k-delta,应把k-delta插入Treap

对于A_k,将基准值delta增加k。对于S_k,将基准值delta减少k,然后在Treap中删除所有小于(lowbound-delta)的元素。

由于我们总是查询第k多的工资,我们可以依照例1的方法,求(总数-k+1)小的工资。我们也不妨换种思路,把Treap建立成一个关键字反序大小比较的Treap,即在比较函数中规定如果a>b(实际的),则a小于b(逻辑的)a放在b的左子树。这虽然难以理解,但却能够满足一定的逻辑顺序。这样建立出的Treap就是自然的从大到小排序的了,查询第k多的工资,就是求排名第k的元素。

orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz orz 

代码:

#include 
#include 
#include 
struct data
{
    int l , r , size , rnd ,v ;
}tr[100005] ;
int size , tmp , root , leave , n , m;
void update(int &k)
{
    tr[k].size = tr[tr[k].l].size + tr[tr[k].r].size + 1 ;
}
void lturn(int &k)
{
    int t = tr[k].r ;
    tr[k].r = tr[t].l ;
    tr[t].l = k ;
    tr[t].size = tr[k].size ;
    update(k) ;
    k = t ;
}
void rturn(int &k)
{
    int t = tr[k].l ;
    tr[k].l = tr[t].r ;
    tr[t].r = k ;
    tr[t].size = tr[k].size ;
    update(k) ;
    k = t ;
}
void insert(int &k , int x)
{
    if(k == 0)
    {
        size ++ ;
        k = size ;
        tr[k].size =  1 ;
        tr[k].v = x ;
        tr[k].rnd = rand() ;
        return ;
    }
    tr[k].size ++ ;
    if(x > tr[k].v)
    {
        insert(tr[k].r , x) ;
        if(tr[tr[k].r].rnd < tr[k].rnd) lturn(k) ;
    }else
    {
        insert(tr[k].l , x) ;
        if(tr[tr[k].l].rnd < tr[k].rnd) rturn(k) ;
    }
}
int del(int &k , int x)
{
    int t ;
    if(k == 0) return 0 ;
    if(tr[k].v < x)
    {
        t = 1 + tr[tr[k].l].size , k = tr[k].r ; return t + del(k , x) ;
    }else
    {
        t = del(tr[k].l , x) , tr[k].size -= t ; return t ;
    }
}
int find(int k , int x)
{
    if(tr[tr[k].l].size + 1 == x) return tr[k].v + tmp ;
    else if(tr[tr[k].l].size + 1 < x) return find(tr[k].r , x - tr[tr[k].l].size - 1) ;
    else return find(tr[k].l , x) ;
}
int main()
{
    scanf("%d%d" , &n , &m) ;
    char s[1] ;
    int x ;
    while(n--)
    {
        scanf("%s%d" , s , &x) ;
        if(s[0] == 'I') if(x >= m)insert(root , x-tmp) ;
        if(s[0] == 'A') tmp += x ;
        if(s[0] == 'S')
        {
            tmp -= x ;
            leave += del(root , m - tmp) ; 
        }
        if(s[0] == 'F')
        {
            if(x > tr[root].size) printf("-1\n") ;
            else printf("%d\n" , find(root , tr[root].size - x +1)) ;
        }
    }
    printf("%d\n" , leave) ;
}

你可能感兴趣的:(Balanced,Binary,Tree,Treap,BZOJ刷题录)