洛谷 P1486 [NOI2004]郁闷的出纳员【Treap】题解+AC代码

题目描述

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

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

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

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

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

输入格式：

第一行有两个非负整数n和min。n表示下面有多少条命令，min表示工资下界。

接下来的n行，每行表示一条命令。命令可以是以下四种之一：

名称 格式 作用

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

A命令 A_k 把每位员工的工资加上k

S命令 S_k 把每位员工的工资扣除k

F命令 F_k 查询第k多的工资

_（下划线）表示一个空格，I命令、A命令、S命令中的k是一个非负整数，F命令中的k是一个正整数。

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

输出格式：

输出文件的行数为F命令的条数加一。

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

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

输入样例#1：

9 10

I 60

I 70

S 50

F 2

I 30

S 15

A 5

F 1

F 2

输出样例#1：

10

20

-1

2

说明

I命令的条数不超过100000

A命令和S命令的总条数不超过100

F命令的条数不超过100000

每次工资调整的调整量不超过1000

新员工的工资不超过100000

---

也是比较裸的一题

说一下几个坑点吧

• 初始工资低于标准直接离开的人不计入离开的人总数(为这个找了好久 = =)
• 删除时记得更新总人数要用该结点的cnt，不要直接- -或++

还有不要吐槽蒟蒻神奇的删除方式

(先记录待删除权值再查找删除什么的 = =)

嗯更新工资也是

蒟蒻直接暴力把所有结点都递归更新了一遍

反正思路特直观就对了= =

---

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

void print(int x)
{
    if(x<0){putchar('-');x=-x;}
    if(x>9)print(x/10);
    putchar(x%10+'0');
}

int n,minn;
int num,ans;
queue<int> q;
struct node
{
    node* ch[2];
    int v,r,sum,cnt;
    node(int v) :v(v) {r=rand();sum=cnt=1;ch[0]=ch[1]=NULL;};
    int cmp(int x){if(x==v)return -1;return x<v?0:1;}
    void update()
    {
        sum=cnt;
        if(ch[0]!=NULL) sum+=ch[0]->sum;
        if(ch[1]!=NULL) sum+=ch[1]->sum;
    }
};
node* rt=NULL;

void rotate(node* &p,int d)
{
    node* k=p->ch[d^1];
    p->ch[d^1]=k->ch[d];
    k->ch[d]=p;
    p->update(); k->update();
    p=k;
}

void ins(node* &p,int x)
{
    if(p==NULL){p=new node(x);return;}
    if(x==p->v){p->sum++;p->cnt++;return;}
    int d=p->cmp(x);
    ins(p->ch[d],x);
    if(p->ch[d]->r < p->r)rotate(p,d^1);
    p->update();
}

void up(node* &p,int x)
{
    if(p==NULL) return ;
    p->v+=x;
    if(p->v<minn)
    {
        q.push(p->v);
        int tot=p->cnt;
        ans+=tot;num-=tot;
        //总人数的更新用p->cnt
    }
    if(p->ch[0]!=NULL)up(p->ch[0],x);
    if(p->ch[1]!=NULL)up(p->ch[1],x);
}

void del(node* &p,int x){
    if(p==NULL) return;
    if(x==p->v)
    {
        if(p->ch[0]==NULL){node *k=p; p=p->ch[1]; delete(k); }
        else if(p->ch[1]==NULL){node *k=p; p=p->ch[0]; delete(k); }
        else
        {
            int dd=p->ch[0]->r < p->ch[1]->r ?1 :0;
            rotate(p,dd); del(p->ch[dd],x);
        }
    }

    else if(x < p->v)del(p->ch[0],x);
    else del(p->ch[1],x);
    if(p!=NULL)p->update();
}

int kth(node* p,int x)
{
    //找第k大同之前发过的模板，就不注释了
    int sum=0;
    if(p->ch[1]!=NULL) sum=p->ch[1]->sum;
    if(x<=sum) return kth(p->ch[1],x);
    else if(x<=sum+p->cnt) return p->v;
    else return kth(p->ch[0],x-sum-p->cnt);
}

int main()
{
    n=read();minn=read();
    for(int i=1;i<=n;i++)
    {
        char ss; scanf("%s",&ss);
        int x=read();
        if(ss=='I')
        {
            if(x<minn)continue;//这里不用计入离开总人数
            else ins(rt,x),num++;
        }
        else if(ss=='A') up(rt,x);
        else if(ss=='S') up(rt,-x);
        else if(ss=='F')
        {
            if(x>num) print(-1);
            else print(kth(rt,x));
            printf("\n");
        }
        while(!q.empty()) {del(rt,q.front());q.pop();}
        //把之前记录的权值都删除
    }
    print(ans);
    return 0;
}