BZOJ1503[NOI2004]郁闷的出纳员——treap

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

这道题是一道经典的平衡树练习题,题中一共有四种操作:插入,查询,修改(上调和下降)。

这里面唯一和一般的treap不一样的操作就是修改,如果每一次都把每个数加上或减掉一个值显然是不行的,但我们发现每次修改都是对全体修改,那么我们就可以不进行修改,相对的把工资下界调整,这样也就相当于修改了每个员工的工资。

但是如果在某次涨工资之后来了一名员工,他并没有赶上涨工资,如果他的工资大于初始工资下界(这样才能进来)但却小于当前工资下界,那么下一次降工资后他本来不会跳槽(下降之后仍大于初始工资下界),却因为小于当前工资下界,那他到底会不会跳槽呢?依据题意是不会的,但如果按上述做法就是会跳槽的。那么就要在他刚进公司时我们要先给他加上之前涨的工资(使他工资一定大于当前工资下界),在查询到他时再减掉就可以了。

工资下降相当于工资下界上调,这样就有可能有人会跳槽,因为I命令不超过10^5,树中最多不会超过10^5个节点,删除点个数也不会超过这些,所以暴力删除就好。

如果查前驱函数是void函数要记得每次查询之后把answer归零。

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<ctime>
#include<cstring>
using namespace std;
int r[100010];
int v[100010];
int w[100010];
int ls[100010];
int rs[100010];
int size[100010];
int tot;
int root;
int n,m;
int del;
int mini;
int x;
char a[10];
int answer;
int ans;
void up(int x)
{
    size[x]=size[rs[x]]+size[ls[x]]+w[x];
}
void rturn(int &x)
{
    int t;
    t=ls[x];
    ls[x]=rs[t];
    rs[t]=x;
    size[t]=size[x];
    up(x);
    x=t;
}
void lturn(int &x)
{
    int t;
    t=rs[x];
    rs[x]=ls[t];
    ls[t]=x;
    size[t]=size[x];
    up(x);
    x=t;
}
void insert_sum(int x,int &i)
{
    if(!i)
    {
        i=++tot;
        w[i]=size[i]=1;
        v[i]=x;
        r[i]=rand();
        return ;
    }
    size[i]++;
    if(x==v[i])
    {
        w[i]++;
    }
    else if(x>v[i])
    {
        insert_sum(x,rs[i]);
        if(r[rs[i]]<r[i])
        {
            lturn(i);
        }
    }
    else
    {
        insert_sum(x,ls[i]);
        if(r[ls[i]]<r[i])    
        {
            rturn(i);
        }
    }
      
    return ;
}
void delete_sum(int x,int &i)
{
    if(i==0)
    {
        return ;
    }
    if(v[i]==x)
    {
        if(w[i]>1)
        {
            w[i]--;
            size[i]--;
            return ;
        }
        if((ls[i]*rs[i])==0)
        {
            i=ls[i]+rs[i];
        }
        else if(r[ls[i]]<r[rs[i]])
        {
            rturn(i);
            delete_sum(x,i);
        }
        else
        {
            lturn(i);
            delete_sum(x,i);
        }
        return ;
    }
    size[i]--;
    if(v[i]<x)
    {
        delete_sum(x,rs[i]);
    }
    else
    {
        delete_sum(x,ls[i]);
    }
    return ;
}
int ask_sum(int x,int i)
{
    if(i==0)
    {
        return 0;
    }
    if(x>size[ls[i]]+w[i])
    {
        return ask_sum(x-size[ls[i]]-w[i],rs[i]);
    }
    else if(size[ls[i]]>=x)
    {
        return ask_sum(x,ls[i]);
    }
    else
    {
        return v[i];
    }
}
void ask_front(int x,int i)
{
    if(i==0)
    {
        return ;
    }
    if(v[i]<x)
    {
        answer=i;
        ask_front(x,rs[i]);
        return ;
    }
    else
    {
        ask_front(x,ls[i]);
        return ;
    }
    return ;
}
int main()
{
    srand(12378);
    scanf("%d%d",&n,&m);
    mini=m;
    for(int i=1;i<=n;i++)
    {
        scanf("%s",a+1);
        scanf("%d",&x);
        if(a[1]=='I')
        {
            if(x>=mini)
            {
                insert_sum(x-del,root);
            }
        }
        else if(a[1]=='A')
        {
            m-=x;
            del+=x;
        }
        else if(a[1]=='S')
        {
            m+=x;
            del-=x;
            answer=0;
            ask_front(m,root);
            while(answer!=0)
            {
                delete_sum(v[answer],root);
                ans++;
                answer=0;
                ask_front(m,root);
                 
            }
        }
        else if(a[1]=='F')
        {
            if(size[root]-x+1>0)
            {
                printf("%d\n",ask_sum(size[root]-x+1,root)+del);
            }
            else
            {
                printf("-1\n");
            }
        }
    }
    printf("%d",ans);
}
posted @ 2018-05-03 20:04  The_Virtuoso  阅读(386)  评论(0编辑  收藏  举报