bzoj1503 郁闷的出纳员

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

分析:
普通的splay…
不过要加一个变量w ,然后每次A操作时w+=num.
S操作:w-=num,且把minn-w(最低工资)在树中的前驱旋到根之后把离开公司总人数sum加上该点元素个数+左子树元素个数,同时更新root.
I操作:如果员工进入时工资x就低于min则直接踢出,且不算入离开公司总人数;
否则把x-w加入树中.

这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>

using namespace std;

const int N=200001;
int ch[N][2],pre[N],size[N],cnt[N],v[N];
int top,root,tot=0,w,lim,ans,n,minn;  //w记录加减情况 

void clear(int x)
{
    ch[x][0]=ch[x][1]=pre[x]=size[x]=cnt[x]=v[x]=0;
}

bool get(int x)
{
    return ch[pre[x]][1]==x;  //x是不是爸爸的右儿子 
}

void update(int x)
{
    size[x]=cnt[x];
    if (ch[x][0]) size[x]+=size[ch[x][0]];
    if (ch[x][1]) size[x]+=size[ch[x][1]];
}

void rotate(int x)
{
    int fa=pre[x];
    int grand=pre[fa];
    int wh=get(x);
    ch[fa][wh]=ch[x][wh^1];
    pre[ch[fa][wh]]=fa;
    ch[x][wh^1]=fa;
    pre[fa]=x;
    pre[x]=grand;
    if (grand)
    {
        ch[grand][ch[grand][1]==fa ? 1:0]=x;
    }
    update(fa); update(x);  //////
    return;
}

void splay(int x)
{
    for (int fa;fa=pre[x];rotate(x))
        if (pre[fa])
           rotate(get(x)==get(fa)? fa:x);
    root=x;
}

void insert(int x)
{
    if (root==0)
    {
        top++;
        cnt[top]=1; ch[top][0]=ch[top][1]=pre[top]=0;
        size[top]=1; cnt[top]=1; v[top]=x;
        root=top;
        return;
    }
    int now=root,fa=0;
    while (1)
    {
        if (v[now]==x)
        {
            cnt[now]++;
            update(now);
            update(fa);
            splay(now);
            return;
        }
        fa=now;
        now=ch[now][x>v[now]];
        if (now==0)
        {
            top++;
            ch[top][0]=ch[top][1]=0;
            pre[top]=fa;
            size[top]=1;
            cnt[top]=1; v[top]=x;
            ch[fa][x>v[fa]]=top;   //
            update(fa);
            splay(top);
            return;
        }
    }
}

int find(int x)
{
    int now=root;
    while (1)
    {
        if (ch[now][0]&&x<=size[ch[now][0]]) now=ch[now][0];
        else{
            int temp=(ch[now][0]?size[ch[now][0]]:0)+cnt[now];
            if (x<=temp) 
               return v[now];
            x-=temp; now=ch[now][1];
        }
    }
}

void fro(int k,int x)  //找前驱 
{
    if (!k) return;
    if (v[k]<x) {
        ans=k;
        fro(ch[k][1],x);
    }
    else fro(ch[k][0],x);
}

int main()
{
    scanf("%d%d",&n,&minn);
    for (int i=1;i<=n;i++)
    {
        char cha,c;
        int opt;
        scanf("%c",&c);
        scanf("%c%d",&cha,&opt);
        if (cha=='I')
        {
            if (opt>=minn) insert(opt-w);  //差值 
        }
        else if (cha=='A')
        {
            w+=opt; 
        } 
        else if (cha=='S')
        {
            w-=opt;
            lim=minn-w;  //把minn-w的在树中的前驱旋到根
            ans=1000000;
            fro(root,lim);  //前驱 
            if (ans!=1000000)
            {
                splay(ans);
                tot+=size[ch[root][0]]+cnt[root];
                root=ch[root][1];
                pre[root]=0;
                update(root);   
            } 
        }
        else {
            int sz=size[root];
            if (opt>sz) printf("-1\n");
            else printf("%d\n",find(sz-opt+1)+w);
            //找第k大->找排名sz-k+1的值 
        } 
    }
    printf("%d\n",tot);
    return 0;
} 
posted @ 2017-05-31 07:28  wtt3117  阅读(71)  评论(0编辑  收藏  举报