BZOJ1503:[NOI2004]郁闷的出纳员——题解
http://www.lydsy.com/JudgeOnline/problem.php?id=1503
(题面复制的洛谷的,因为洛谷好看)
题目描述
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
————————————————————————————
维护一棵平衡树,记录一个det记录这棵树上的点最后需要加上det才为真正的值。
于是我们每次插入的人的钱都是k-det。
当我们查询哪些人要被删的时候,我们插入minn-det,然后删掉该点的左子树即可。
那么询问第k大就暴力在平衡树搜即可(方法基本同线段树的搜索)
#include<cstdio> #include<queue> #include<cctype> #include<cstring> #include<cmath> #include<vector> #include<algorithm> using namespace std; const int N=100001; inline int read(){ int X=0,w=0;char ch=0; while(!isdigit(ch)){w|=ch=='-';ch=getchar();} while(isdigit(ch))X=(X<<3)+(X<<1)+(ch^48),ch=getchar(); return w?-X:X; } int fa[N],tr[N][2],key[N],cnt[N],size[N]; int root,sz; inline void clear(int x){ fa[x]=tr[x][0]=tr[x][1]=key[x]=cnt[x]=size[x]=0; return; } inline bool get(int x){ return tr[fa[x]][1]==x; } inline void update(int x){ if(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]]; } return; } inline void rotate(int x){ int old=fa[x],oldf=fa[old],which=get(x); tr[old][which]=tr[x][which^1];fa[tr[old][which]]=old; fa[old]=x;tr[x][which^1]=old;fa[x]=oldf; if(oldf)tr[oldf][tr[oldf][1]==old]=x; update(old);update(x); return; } inline void splay(int x){ int f=fa[x]; while(f){ if(fa[f])rotate((get(x)==get(f)?f:x)); rotate(x);f=fa[x]; } root=x; return; } inline void insert(int v){ if(!root){ sz++;tr[sz][0]=tr[sz][1]=fa[sz]=0; key[sz]=v;cnt[sz]=size[sz]=1;root=sz; return; } int now=root,f=0; while(233){ if(key[now]==v){ cnt[now]++;update(now);update(f);splay(now); break; } f=now; now=tr[now][key[now]<v]; if(!now){ sz++;tr[sz][0]=tr[sz][1]=0;fa[sz]=f; key[sz]=v;cnt[sz]=size[sz]=1; tr[f][key[f]<v]=sz; update(f);splay(sz); break; } } return; } inline void del(){ if(cnt[root]>1){ cnt[root]--; size[root]--; return; } if(!tr[root][1]){ clear(root);root=0;return; } int oldroot=root;root=tr[root][1];fa[root]=0;clear(oldroot); return; } int query(int k,int x){ if(size[tr[x][1]]<k&&k<=cnt[x]+size[tr[x][1]])return key[x]; else if(size[tr[x][1]]<k)return query(k-size[tr[x][1]]-cnt[x],tr[x][0]); else return query(k,tr[x][1]); } int main(){ int n=read(); int minn=read(); int det=0; int delnum=0; for(int i=1;i<=n;i++){ char ch=0; while(ch<'A'||ch>'Z')ch=getchar(); int k=read(); if(ch=='I')if(k>=minn)insert(k-det); if(ch=='A')det+=k; if(ch=='S'){ det-=k; insert(minn-det); size[root]-=size[tr[root][0]]; delnum+=size[tr[root][0]]; clear(tr[root][0]); tr[root][0]=0; del(); } if(ch=='F'){ if(size[root]<k)puts("-1"); else{ printf("%d\n",query(k,root)+det); } } } printf("%d\n",delnum); return 0; }