【splay】BZOJ 1503 郁闷的出纳员

BZOJ 1503 郁闷的出纳员


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


Solution

splay貌似已经写着越来越简单了..
普通的splay…
不过要加一个变量S
每次全扣工资x,则S-=x.
加x的话S+=x.
每次当加入一个新数字x就相当较以前加入一个x-S的数字,这样每次全加全扣的时候就不用再打tag什么的了..

然后每次A操作时S-=x.
S操作S+=x且把min-S的在树中的前驱旋到根之后把离开公司总人数sum加上该点元素个数+左子树元素个数,同时更新root.
I操作如果员工进入时工资x就低于min则直接踢出且离开公司总人数不加然后再把x-ss加入树中.
查询的时候记得判定没有出去的情况和-1.

Code

#include
#include
#include
#include

using namespace std;

#define maxn 100001

struct splay{
    int ch[2],size,w,v,fa;
}t[maxn];

int root,size,ans;

inline int in()
{
    int x=0;char ch=getchar();
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-'0',ch=getchar();
    return x;
}

inline void update(int k){t[k].size=t[t[k].ch[0]].size+t[t[k].ch[1]].size+t[k].w;}

void rotate(int x,int &k,int d)
{
    int y,z;
    y=t[x].fa,z=t[y].fa;
    if(y==k)k=x;
    else {if(t[z].ch[0]==y)t[z].ch[0]=x;else t[z].ch[1]=x;}
    t[x].fa=z,t[y].fa=x,t[t[x].ch[d]].fa=y;
    t[y].ch[d^1]=t[x].ch[d];t[x].ch[d]=y;
    update(y),update(x);
}

void splay(int x,int &k)
{
    int y,z;
    while(x!=k)
    {
        y=t[x].fa,z=t[y].fa;
        if(y!=k)
        {
            if(t[z].ch[0]==y&&t[y].ch[0]==x)rotate(y,k,1);
            else if(t[z].ch[1]==y&&t[y].ch[1]==x)rotate(y,k,0);
            else if(t[z].ch[0]==y&&t[y].ch[1]==x)rotate(x,k,0);
            else if(t[z].ch[1]==y&&t[y].ch[0]==x)rotate(x,k,1);
        }
        if(t[t[x].fa].ch[0]==x)rotate(x,k,1);
        else rotate(x,k,0);
    }
}

void insert(int x,int &k,int last)
{
    if(!k)
    {
        k=++size;
        t[k].size=t[k].w=1;
        t[k].fa=last;
        t[k].v=x;
        splay(k,root);
        return;
    }
    t[k].size++;
    if(t[k].v==x)t[k].w++;
    else if(t[k].v1],k);
    else insert(x,t[k].ch[0],k);
}

void get_pre(int k,int x)
{
    if(!k)return;
    if(t[k].v1],x);}
    else get_pre(t[k].ch[0],x);
}

int get_x(int k,int x)
{
    if(x<=t[t[k].ch[1]].size)return get_x(t[k].ch[1],x);
    else if(x>t[t[k].ch[1]].size+t[k].w)return get_x(t[k].ch[0],x-t[k].w-t[t[k].ch[1]].size);
    else return k;
}

int main()
{
    int n,m,ss=0,an=0;
    freopen("1503.in","r",stdin);
    n=in(),m=in();
    for(int i=1;i<=n;i++)
    {
        char ch;
        int x;
        scanf("\n%c%d",&ch,&x);
        if(ch=='I')
        {
            if(m>x)continue;
            else
            {
                x-=ss;
                insert(x,root,0);
            }
        }
        else if(ch=='A')
        {
            ss+=x;
        }
        else if(ch=='S')
        {
            ss-=x;
            ans=9999999;
            get_pre(root,m-ss);
            if(ans!=9999999)
            {
                splay(ans,root);
                an+=t[t[root].ch[0]].size+t[root].w;
                root=t[root].ch[1];
                update(root);
            }
        }
        else {
            if(x>t[root].size){printf("-1\n");continue;}
            int x1=get_x(root,x);
            printf("%d\n",t[x1].v+ss);
            splay(x1,root);
        }
    }
    printf("%d\n",an);
}

你可能感兴趣的:(数据结构,splay,splay,数据结构)