BZOJ1503[NOI2004]郁闷的出纳员——treap
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。
输出文件的最后一行包含一个整数,为离开公司的员工的总数。
9 10 I 60 I 70 S 50 F 2 I 30 S 15 A 5 F 1 F 2
10 20 -1 2
说明
I命令的条数不超过100000
A命令和S命令的总条数不超过100
F命令的条数不超过100000
每次工资调整的调整量不超过1000
新员工的工资不超过100000
这道题是一道经典的平衡树练习题,题中一共有四种操作:插入,查询,修改(上调和下降)。
这里面唯一和一般的treap不一样的操作就是修改,如果每一次都把每个数加上或减掉一个值显然是不行的,但我们发现每次修改都是对全体修改,那么我们就可以不进行修改,相对的把工资下界调整,这样也就相当于修改了每个员工的工资。
但是如果在某次涨工资之后来了一名员工,他并没有赶上涨工资,如果他的工资大于初始工资下界(这样才能进来)但却小于当前工资下界,那么下一次降工资后他本来不会跳槽(下降之后仍大于初始工资下界),却因为小于当前工资下界,那他到底会不会跳槽呢?依据题意是不会的,但如果按上述做法就是会跳槽的。那么就要在他刚进公司时我们要先给他加上之前涨的工资(使他工资一定大于当前工资下界),在查询到他时再减掉就可以了。
工资下降相当于工资下界上调,这样就有可能有人会跳槽,因为I命令不超过10^5,树中最多不会超过10^5个节点,删除点个数也不会超过这些,所以暴力删除就好。
如果查前驱函数是void函数要记得每次查询之后把answer归零。
#include<cstdio> #include<algorithm> #include<iostream> #include<cmath> #include<ctime> #include<cstring> using namespace std; int r[100010]; int v[100010]; int w[100010]; int ls[100010]; int rs[100010]; int size[100010]; int tot; int root; int n,m; int del; int mini; int x; char a[10]; int answer; int ans; void up(int x) { size[x]=size[rs[x]]+size[ls[x]]+w[x]; } void rturn(int &x) { int t; t=ls[x]; ls[x]=rs[t]; rs[t]=x; size[t]=size[x]; up(x); x=t; } void lturn(int &x) { int t; t=rs[x]; rs[x]=ls[t]; ls[t]=x; size[t]=size[x]; up(x); x=t; } void insert_sum(int x,int &i) { if(!i) { i=++tot; w[i]=size[i]=1; v[i]=x; r[i]=rand(); return ; } size[i]++; if(x==v[i]) { w[i]++; } else if(x>v[i]) { insert_sum(x,rs[i]); if(r[rs[i]]<r[i]) { lturn(i); } } else { insert_sum(x,ls[i]); if(r[ls[i]]<r[i]) { rturn(i); } } return ; } void delete_sum(int x,int &i) { if(i==0) { return ; } if(v[i]==x) { if(w[i]>1) { w[i]--; size[i]--; return ; } if((ls[i]*rs[i])==0) { i=ls[i]+rs[i]; } else if(r[ls[i]]<r[rs[i]]) { rturn(i); delete_sum(x,i); } else { lturn(i); delete_sum(x,i); } return ; } size[i]--; if(v[i]<x) { delete_sum(x,rs[i]); } else { delete_sum(x,ls[i]); } return ; } int ask_sum(int x,int i) { if(i==0) { return 0; } if(x>size[ls[i]]+w[i]) { return ask_sum(x-size[ls[i]]-w[i],rs[i]); } else if(size[ls[i]]>=x) { return ask_sum(x,ls[i]); } else { return v[i]; } } void ask_front(int x,int i) { if(i==0) { return ; } if(v[i]<x) { answer=i; ask_front(x,rs[i]); return ; } else { ask_front(x,ls[i]); return ; } return ; } int main() { srand(12378); scanf("%d%d",&n,&m); mini=m; for(int i=1;i<=n;i++) { scanf("%s",a+1); scanf("%d",&x); if(a[1]=='I') { if(x>=mini) { insert_sum(x-del,root); } } else if(a[1]=='A') { m-=x; del+=x; } else if(a[1]=='S') { m+=x; del-=x; answer=0; ask_front(m,root); while(answer!=0) { delete_sum(v[answer],root); ans++; answer=0; ask_front(m,root); } } else if(a[1]=='F') { if(size[root]-x+1>0) { printf("%d\n",ask_sum(size[root]-x+1,root)+del); } else { printf("-1\n"); } } } printf("%d",ans); }