【NOI2004】郁闷的出纳员

Description

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

Input

第一行有两个非负整数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是一个正整数。 在初始时,可以认为公司里一个员工也没有。

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

NOI2004 ,平衡树

 

题解:

第三道splay板子训练题,因为对两个改值的操作都是对总体进行修改,我们可以用splay维护一个对于总体加减的相对值。记相对值为data

所以加入节点时真正的是总值-相对值及 x-data,然后对于插入上升操作只要将data加上x就可以了,删除操作稍微推一下,因为vi+data<minn(vi为相对值,minn为下限)为不合法,所以我们只要找到大小为minn-data的节点,所以把他的左子树切断,以后就不会再访问左子树了,最后一个操作我们可以到着想区间第k大可以用(总数-k+1)小来表达。所以splay维护一下就好了。

 

代码:

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<cstring>
const int MAXN=100100;
using namespace std;
int tr[MAXN][2],cnt[MAXN],size[MAXN],pr[MAXN],key[MAXN];
int roof=0,tot=0;
int n,minn,data=0,ans=0;
 
int gi() {
  int res=0,f=1;
  char ch=getchar();
  while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
  if(ch=='-') ch=getchar(),f=-1;
  while(ch>='0'&&ch<='9') res=res*10+ch-'0',ch=getchar();
  return res*f;
}
 
void newnode(int k,int fu){
    key[++tot]=k;cnt[tot]=size[tot]=1;tr[tot][0]=tr[tot][1]=0;pr[tot]=fu;
}
 
void cl(int x){
    key[x]=cnt[x]=size[x]=tr[x][0]=tr[x][1]=pr[x]=0;
}
 
void update(int x){
    size[x]=cnt[x];
    if(tr[x][0]) size[x]+=size[tr[x][0]];
    if(tr[x][1]) size[x]+=size[tr[x][1]];
}
 
void Rotate(int x,int kind){
    int y=pr[x];
    tr[y][!kind]=tr[x][kind];
    pr[tr[x][kind]]=y;
    if(pr[y]) tr[pr[y]][tr[pr[y]][1]==y]=x;
    pr[x]=pr[y];
    tr[x][kind]=y;
    pr[y]=x;
    update(y);
    update(x);
}
 
 
void splay(int x,int goal){
    while(pr[x]!=goal){
        if(pr[pr[x]]==goal) 
        Rotate(x,tr[pr[x]][0]==x);
        else{
            int y=pr[x],kind=tr[pr[y]][0]==y;
            if(tr[y][kind]==x){
                Rotate(x,!kind);
                Rotate(x,kind);
            }
            else{
                Rotate(y,kind);
                Rotate(x,kind);
            }
        }
    }
    if(goal==0) roof=x;
}
 
void insert(int k){
    if(roof==0){
        newnode(k,0);roof=tot;return;
    }
    int now=roof,fu=0;
    while(1){
        if(key[now]==k){
            cnt[now]++,update(now),update(pr[now]);
            splay(now,0);return;
        }
        fu=now,now=tr[now][k>key[now]];
        if(now==0){
            newnode(k,fu);
            tr[fu][k>key[fu]]=tot;
            update(fu);
            splay(tot,0);
            return;
        }
    }
}
    
 int find(int x) {
  int now=roof;int tmp;
  while(1) {
    if(tr[now][0]&&x<=size[tr[now][0]]) now=tr[now][0];
    else {
      tmp = (tr[now][0] ? size[tr[now][0]] : 0) + cnt[now];
      if(x<=tmp) return key[now];
      x-=tmp;now=tr[now][1];
    }
  }
}
 
void del() {
  if(roof==0) return;
  int k=minn-data;
  int now=roof,st=-1;
  while(1) {
    if(key[now]>=k) {
      if(st!=-1) st = key[now] < key[st] ? now : st;
      else st=now;
    }
    now=tr[now][k > key[now]];
    if(now==0) {
      if(st!=-1) {
        splay(st,0);
        size[st] -= size[tr[st][0]];
        ans+=size[tr[st][0]];
        tr[st][0]=0;
      }
      else {
        ans+= size[roof];
        cl(roof);
        roof=0;
      }
      return;
    }
  }
}
int main(){
    scanf("%d%d",&n,&minn);
    for(int i=1;i<=n;i++){
        int x;char id;
        scanf("%c%c%d",&id,&id,&x);
        if(id=='I') 
        if(x>=minn) insert(x-data);if(id=='A') data+=x;
        if(id=='S') data-=x,del();
        if(id=='F') 
        if(x>size[roof]) printf("-1\n");
        else printf("%d\n",find(size[roof]-x+1)+data);
    }
    printf("%d\n",ans);
}

 

posted @ 2017-08-03 12:33  人间失格—太宰治  阅读(675)  评论(0编辑  收藏  举报