treap题目

题目描述

OIER 公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把当前在公司的所有员工的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。

工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。

老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第 kk 多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。

好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难吧？

如果某个员工的初始工资低于最低工资标准，那么将不计入最后的答案内。

输入格式

第一行有两个整数 nn 和 \minmin。nn 表示下面有多少条命令，\minmin 表示工资下界。

接下来的 nn 行，每行一个字符 xx 和一个整数 kk，表示一条命令。命令可以是以下四种之一：

• I k 新建一个工资档案，初始工资为 kk。如果某员工的初始工资低于工资下界，他将立刻离开公司。

• A k 把每位员工的工资加上 kk 。

• S k 把每位员工的工资扣除 kk。

• F k 查询第 kk 多的工资。

在初始时，可以认为公司里一个员工也没有。

输出格式

对于每条 F 命令，你的程序要输出一行，仅包含一个整数，为当前工资第 kk 多的员工所拿的工资数，如果 kk 大于目前员工的数目，则输出 -1−1。

输出的最后一行包含一个整数，为离开公司的员工的总数。

请注意，初始工资低于工资下界的员工不算做离开公司的员工。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=3e5+5,INF=1e9+5;
int root;
struct node{
    int size,cnt,val,key,l,r;
}tr[N];
int idx=0;
void pushup(int p)
{
    tr[p].size=tr[tr[p].l].size+tr[tr[p].r].size+tr[p].cnt;
}
int get_node(int key)
{
    int p=++idx;
    tr[p].cnt=1;
    tr[p].size=1;
    tr[p].val=rand();
    tr[p].key=key;
    return p;
};
void zig(int &p)
{
    int q=tr[p].l;
    tr[p].l=tr[q].r,tr[q].r=p;p=q;
    pushup(tr[p].r);pushup(p);
}
void zag(int &p)
{
    int q=tr[p].r;
    tr[p].r=tr[q].l,tr[q].l=p,p=q;
    pushup(tr[p].l);pushup(p);
}
void build( )
{
    get_node(-INF),get_node(INF);
    root=1,tr[1].r=2;
    pushup(1);
    if(tr[2].val>tr[1].val) zag(root);
}
void insert(int &p,int key)
{
    if(!p) p=get_node(key);
    else if(tr[p].key==key) tr[p].cnt++;
    else
    {
        if(tr[p].key>key)
        {
            insert(tr[p].l,key);
            if(tr[tr[p].l].val>tr[p].val) zig(p);
        }
        else
        {
            insert(tr[p].r,key);
            if(tr[tr[p].r].val>tr[p].val) zag(p);
        }
    }
    pushup(p);
}
void remove(int &p,int key)
{
    if(!p) return ;
    if(tr[p].key==key)
    {
        if(tr[p].cnt>1) tr[p].cnt--;
        else if(tr[p].l||tr[p].r)
        {
            if(!tr[p].r||tr[tr[p].r].val<tr[tr[p].l].val)
            {
                zig(p);
                remove(tr[p].r,key);
            }
            else
            {
                zag(p);
                remove(tr[p].l,key);
            }
        }
        else p=0;
    }
    else if(key<tr[p].key) remove(tr[p].l,key);
    else remove(tr[p].r,key);
    pushup(p);
}
int get_key_by_rank(int p,int rank)
{
    if(!p) return 0;
    if(tr[tr[p].l].size>=rank) return get_key_by_rank(tr[p].l,rank);
    if(tr[tr[p].l].size+tr[p].cnt>=rank) return tr[p].key;
    return get_key_by_rank(tr[p].r,rank-tr[tr[p].l].size-tr[p].cnt);
}
int get_pre(int p,int key)
{
    if(!p) return -INF;
    if(tr[p].key>key) return get_pre(tr[p].l,key);
    return max(tr[p].key,get_pre(tr[p].r,key));
}
int main()
{
    int n,minn;scanf("%d%d",&n,&minn);
    int num=0,sum=0,s=0;
    build();
    for(int i=1;i<=n;i++)
    {
        int x;
        char op[2];scanf("%s%d",op,&x);
        if(*op=='I')
        {
            if(x-sum>=minn) insert(root,x-sum),num++,s++;
        }
        else if(*op=='A') sum+=x,minn-=x;
        else if(*op=='S')
        {
            sum-=x,minn+=x;
            int a1=minn-1;
            while((a1=get_pre(root,a1))!=-INF)
            {
                remove(root,a1);
                num--;
            }
        }
        else
        {
            if(num<x) puts("-1");
            else
            printf("%d\n",get_key_by_rank(root,num-x+2)+sum);
        }
    }
    printf("%d\n",s-num);
}