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

时间:2021-09-15 09:45:31

题目描述

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