郁闷的收纳员

蒟蒻刚学旋转Treap,打个板子练练手。

题目描述

原题来自:NOI 2004

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

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

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

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

输入格式

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

接下来的 nnn 行,每行表示一个命令。命令可以是以下四种之一:

名称 格式 作用
I 命令 I k 新建一个工资档案,初始工资为 kkk。如果某员工的初始工资低于工资下界,他将立刻离开公司。
A 命令 A k 把每位员工的工资加上 kkk。
S 命令 S k 把每位员工的工资扣除 kkk。
F 命令 F k 查询第 kkk 多的工资。

I 命令,A 命令,S 命令中,kkk 是一个非负整数。F 命令中,kkk 是一个正整数。

在初始时,可以认为公司中一位员工也没有。

输出格式

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

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

输出文件的最后一行包含一个整数,为离开公司的员工的总数(不包括因员工的初始工资低于工资下界而离开公司的情况)。

样例

样例输入

9 10
I 60
I 70
S 50
F 2
I 30
S 15
A 5
F 1
F 2

样例输出

10
20
-1
2

数据范围与提示

对于全部数据,I 命令的条数不超过 10^5A 命令和 S 命令的总条数不超过 100,F 命令的条数不超过 10^5。

每次工资调整的调整量不超过 10^3,新员工的工资不超过 10^5​​。

辣鸡题解:

区间修改,可以不必维护,我们只要相对大小。考虑定义line为零工资线(物理重力势能滑稽)。

A x 则所有已入职员工工资相对line增大,即line-=x;  然后注意保持相对大小,此时权值为x+line(因为line是零啊喂)

S x line操作同理,不同在于S会有员工离职,我们需要将这些点删除。

注意题面:S命令不超过100,最坏情况1e8可以试试,暴搜a!

直接从rt进行dfs   n,回溯时调用del()   logn。     删除复杂度O(nlogn)

注意:

  • 有相同工资的,cnt要减成零(因为我直接cnt次logn调用了)
  • 读题a,如果I时x低于MIN则不不能算到答案里
  • 最后别忘相对到绝对-line
#include<cstdio>
#include<cmath>
#include<ctime>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define ls tr[x].ch[0]
#define rs tr[x].ch[1]
#define ds tr[x].ch[d]
#define reg register
#define inf 0x6fffffff
#define F(i,a,b) for(i=a;i<=b;++i)
using namespace std;
int cnt,rt,line,MIN;
struct TREAP{
    int ch[2],w,siz,cnt,rd;
}tr[1000005];
void up(int x)
{
    tr[x].siz=tr[ls].siz+tr[rs].siz+tr[x].cnt;
}
void rot(int &x,int d)
{
    int son=ds;
    ds=tr[son].ch[d^1];
    tr[son].ch[d^1]=x;
    up(x);
    up(x=son);
}
void in(int &x,int w)
{
    if(!x)
    {
        x=++cnt;                                        //
        tr[x].siz=tr[x].cnt=1;
        tr[x].w=w;
        tr[x].rd=rand();
        return;
    }
    tr[x].siz++;
    if(tr[x].w==w){
        tr[x].cnt++;return;
    }
    int d=w>tr[x].w;
    in(ds,w);
    if(tr[x].rd>tr[ds].rd) rot(x,d);
}
void del(int &x,int w)
{
    if(!x) return;
    if(tr[x].w==w)
    {
        if(tr[x].cnt>1)
        {
            tr[x].cnt--;
            tr[x].siz--;
            return;
        }
        int d=tr[ls].rd>tr[rs].rd;
        if(ls==0||rs==0) x=ls+rs;
        else rot(x,d),del(x,w);
    }
    else tr[x].siz--,del(tr[x].ch[tr[x].w<w],w);
}
void dfs(int x)
{
    if(ls) dfs(ls);
    if(rs) dfs(rs);
    if(MIN+line>tr[x].w)
    {
        int t=tr[x].cnt;            //
        while(t--)
            del(rt,tr[x].w);
    }
}
int kth(int x,int k)
{
    while(x)
    {
        if(k<=tr[ls].siz) x=ls;
        else if(k>tr[x].cnt+tr[ls].siz) k-=tr[x].cnt+tr[ls].siz,x=rs;
        else return tr[x].w;
    }
}
int main()
{
    char s[5];
    srand(time(0));
    int n;
    scanf("%d%d",&n,&MIN);
    reg int x,num=0;
    while(n--)
    {
        scanf("%s %d",s,&x);
        if(s[0]=='I')
        {
            if(x>=MIN)
                in(rt,x+line),++num;
        }
        else if(s[0]=='A')
            line-=x;
        else if(s[0]=='S')
        {
            line+=x;
            dfs(rt);
        }
        else
        {
            if(tr[rt].siz<x)
                printf("-1\n");
            else
                printf("%d\n",kth(rt,tr[rt].siz-x+1)-line);
        }
    }
    printf("%d",num-tr[rt].siz);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/hzoi-yzh/p/10986451.html