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

1503: [NOI2004]郁闷的出纳员

Time Limit: 5 Sec  Memory Limit: 64 MB
Submit: 7749  Solved: 2725
[Submit][Status][Discuss]

Description

OIER公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资。如果他心情好,就可能把每位员工的工资加上一个相同的量。反之,如果心情不好,就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。工资的频繁调整很让员工反感,尤其是集体扣除工资的时候,一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案。老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第k多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?

Input

Output

输出文件的行数为F命令的条数加一。对于每条F命令,你的程序要输出一行,仅包含一个整数,为当前工资第k多的员工所拿的工资数,如果k大于目前员工的数目,则输出-1。输出文件的最后一行包含一个整数,为离开公司的员工的总数。

Sample Input

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

Sample Output

10
20
-1
2

HINT

I命令的条数不超过100000 A命令和S命令的总条数不超过100 F命令的条数不超过100000 每次工资调整的调整量不超过1000 新员工的工资不超过100000

Source

treap 800ms...裸题
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
char c[1];
int n,mn,x,cnt,delta,root,leave;
int size[100005],num[100005],l[100005],r[100005],rnd[100005];
void updata(int x)
{
    size[x]=size[l[x]]+size[r[x]]+1;
}
void rturn(int &k)
{
    int t=l[k];l[k]=r[t];r[t]=k;updata(k);updata(t);k=t;
}
void lturn(int &k)
{
    int t=r[k];r[k]=l[t];l[t]=k;updata(k);updata(t);k=t;
}
void insert(int &k,int x)
{
    if (!k)
    {
        k=++cnt;
        num[k]=x;
        rnd[k]=rand();
        size[k]=1;
        return;
    }
    size[k]++;
    if (x<num[k])
    {
        insert(l[k],x);
        if (rnd[l[k]]<rnd[k]) rturn(k);
    }
    else
    {
        insert(r[k],x);
        if (rnd[r[k]]<rnd[k]) lturn(k);
    }
}
int del(int &k,int x)
{
    int t;
    if (!k) return 0;
    if (num[k]<x)
    {
        t=size[l[k]]+1;
        k=r[k];
        return t+del(k,x);
    }
    else
    {
        t=del(l[k],x);
        size[k]-=t;
        return t;
    }
}
int find(int k,int x)
{
    if (size[l[k]]+1==x) return num[k]+delta;
    if (size[l[k]]+1>x) return find(l[k],x);
    if (size[l[k]]+1<x) return find(r[k],x-size[l[k]]-1);
}
int main()
{
    scanf("%d%d",&n,&mn);
    while(n--)
    {
        scanf("%s%d",c,&x);
        if (c[0]=='I') 
            if (x>=mn) insert(root,x-delta);
        if (c[0]=='A') delta+=x;
        if (c[0]=='S') 
        {
            delta-=x;
            leave+=del(root,mn-delta);
        }
        if (c[0]=='F')
            if (x>size[root]) printf("-1
");
                else printf("%d
",find(root,size[root]-x+1));
    }
    printf("%d",leave);
    return 0;
}

splay 1300ms

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<set>
using namespace std;
char c[5];
int n,x,mn,root,cnt,delta,leave,tree[100001][2],size[100001],fa[100001],num[100001];
inline void update(int k)
{
    size[k]=size[tree[k][0]]+size[tree[k][1]]+1;
}
inline void rotate(int x,int &k)
{
    int y=fa[x],z=fa[y],l,r;
    l=(tree[y][1]==x);r=l^1;
    if (y==k) k=x;
    else tree[z][tree[z][1]==y]=x;
    fa[x]=z; fa[y]=x; fa[tree[x][r]]=y;
    tree[y][l]=tree[x][r]; tree[x][r]=y;
    update(y); update(x);
}
inline void splay(int x,int &k)
{
    while (x!=k)
    {
        int y=fa[x],z=fa[y];
        if (y!=k)
        {
            if ((tree[y][0]==x)^(tree[z][0]==y)) rotate(x,k); else rotate(y,k);
        }
        rotate(x,k);
    }
}
void insert(int k)
{
     if(!root)
     {
         root=++cnt;
         num[cnt]=k;
         size[cnt]=1;
         return;
              }
     int p=root,z;
     while(p)
     {
             z=p;
             ++size[p];
             if(k<num[p])p=tree[p][0];
             else p=tree[p][1];
             }
     if (num[z]>k) tree[z][0]=++cnt;
     else tree[z][1]=++cnt;
     num[cnt]=k;size[cnt]=1;fa[cnt]=z;
     splay(cnt,root);
}
int del(int &x,int f)
{
    if(!x)return 0;
    int k;
    if(num[x]+delta<mn)
    {
          k=del(tree[x][1],x)+size[tree[x][0]]+1;
          size[tree[x][1]]=size[x]-k;
          x=tree[x][1];fa[x]=f;
    }
    else
    {
         k=del(tree[x][0],x);
         size[x]-=k;
     }
    return k;
}
int find(int k,int x)
{
    if (size[tree[k][1]]>=x) return find(tree[k][1],x);
    if (size[tree[k][1]]+1==x) return num[k];
    return find(tree[k][0],x-size[tree[k][1]]-1);
}
int main()
{
    scanf("%d%d",&n,&mn);
    for (int i=1;i<=n;i++)
    {
        scanf("%s%d",c,&x);
        if (c[0]=='I'&&x>=mn) insert(x-delta);
        if (c[0]=='A') delta+=x;
        if (c[0]=='S')
        {
            delta-=x;
            leave+=del(root,0);
        }
        if (c[0]=='F') printf("%d
",x<=size[root]?find(root,x)+delta:-1);
    }
    printf("%d",leave);
    return 0;
}
原文地址:https://www.cnblogs.com/ws-fqk/p/4668049.html