NOI2004 郁闷的出纳员 [Splay]

题目描述

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

题解

  • 看到题目中的查询第k大,很容易想到使用平衡树,但是本人蒟蒻,只学过splay,不向其他巨佬,花式切题,treap,权值线段树,替罪羊树,vector一个都不会.
  • 操作中,因为A命令和S命令都是对于所有的员工而不是某一特定的区间,所以我们可以在外部统一存储.特别地,对于A操作,我们不需要管,而对于S操作,我们可以找到第一个符合条件也就是(ming-p)(其中p为外部A或S存储的值)的值,然后把它旋转到根,这就需要我们开始时插入一个极大的值,保证可以查询到符合条件的值,进行splay操作后,根节点左子树都是小于它的值,我们把它全部扔掉
#include<bits/stdc++.h>
#define in(i) (i=read())
using namespace std;
typedef long long lol;
lol read() {
    lol ans=0,f=1; char i=getchar();
    while(i<'0' || i>'9') {if(i=='-') f=-1; i=getchar();}
    while(i>='0' && i<='9') {ans=(ans<<1)+(ans<<3)+i-'0'; i=getchar();}
    return ans*f;
}
const int inf=2147483647;
lol n,minx,ans,tot,p,root;
struct node {
    int ff,ch[2],cnt,size,val;
}t[100010];
void push_up(int x) {
    t[x].size=t[t[x].ch[0]].size+t[t[x].ch[1]].size+t[x].cnt;
}
void rorate(int x) {
    int y=t[x].ff,z=t[y].ff;
    int k=(t[y].ch[1]==x);
    t[z].ch[t[z].ch[1]==y]=x; t[x].ff=z;
    t[y].ch[k]=t[x].ch[k^1]; t[t[x].ch[k^1]].ff=y;
    t[x].ch[k^1]=y; t[y].ff=x;
    push_up(y); push_up(x);
}
void splay(int x,int goal) {
    while(t[x].ff!=goal) {
        int y=t[x].ff,z=t[y].ff;
        if(z!=goal) (t[z].ch[0]==y)^(t[y].ch[0]==x)?rorate(x):rorate(y);
        rorate(x);
    }
    if(!goal) root=x;
}
inline void find(int x) {
    int u=root;
    if(!u) return;
    while(t[u].ch[x>t[u].val] && x!=t[u].val) u=t[u].ch[x>t[u].val];
    splay(u,0);
}
inline void insert(int x) {
    int u=root,ff=0;
    while(u && t[u].val!=x) {
        ff=u;
        u=t[u].ch[x>t[u].val];
    }
    if(u) t[u].cnt++;
    else {
        u=++tot;
        if(ff) t[ff].ch[x>t[ff].val]=u;
        t[u].ch[0]=t[u].ch[1]=0;
        t[u].ff=ff; t[u].val=x;
        t[u].size=t[u].cnt=1;
    }
    splay(u,0);
}
inline int Next(int x,int f) {
    find(x); int u=root;
    if(t[u].val>=x && f) return u;
    if(t[u].val<=x && !f) return u;
    u=t[u].ch[f];
    while(t[u].ch[f^1]) u=t[u].ch[f^1];
    return u;
}
inline void Del(int x) {
    int last=Next(x,0),next=Next(x,1);
    splay(last,0); splay(next,last);
    int del=t[next].ch[0];
    if(t[del].cnt>1) {
        t[del].cnt--;
        splay(del,0);
    }
    else t[next].ch[0]=0;
}
int kth(int x) {
    int u=root; 
    if(t[u].size<x) return -inf;
    while(1) {
        int y=t[u].ch[1];
        if(x>t[y].size+t[u].cnt) {
            x-=t[y].size+t[u].cnt;
            u=t[u].ch[0];
        }
        else
            if(x<=t[y].size) u=y;
            else return t[u].val;
    }
}
int main()
{
    //freopen("unhappy.in","r",stdin);
    //freopen("unhappy.out","w",stdout);
    in(n);in(minx);
    int lg=0;
    insert(inf);
    for(lol i=1;i<=n;i++) {
        char ch; lol k;
        cin>>ch; in(k);
        if(ch=='I') {
            if(k>=minx) insert(k-p),lg++;
        }
        else if(ch=='A') p+=k;
        else if(ch=='S') {
            p-=k;
            int f=Next(minx-p,1);
            splay(f,0);
            ans+=t[t[root].ch[0]].size;
            lg-=t[t[root].ch[0]].size;
            t[t[root].ch[0]].size=t[t[root].ch[0]].cnt=0;
            t[root].ch[0]=0;
            push_up(root);
        }
        else if(ch=='F') {
            int x=kth(k+1);
            if(x==-inf) cout<<-1<<endl;
            else cout<<x+p<<endl;
        }
    }
    cout<<ans<<endl;
    return 0;
}

博主蒟蒻,随意转载.但必须附上原文链接

http://www.cnblogs.com/real-l/

原文地址:https://www.cnblogs.com/real-l/p/9337818.html