[BZOJ 1503] [NOI2004] 郁闷的出纳员

1503: [NOI2004] 郁闷的出纳员

Time Limit: 5 SecMemory Limit: 64 MB

Description

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

Input

#include<bits/stdc++.h>
using namespace std;
const int N=300010;
int n,M,tot1,res,ch
[2],fa
,size
,root,k
,det;
inline void pushdown(int x) {
if(x) size[x]=size[ch[x][0]]+size[ch[x][1]]+1;
}
inline void zig(int x) {
int y=fa[x],z=fa[y];
fa[y]=x;fa[x]=z;
ch[y][0]=ch[x][1];fa[ch[x][1]]=y;
ch[x][1]=y;
if(y==ch[z][0]) ch[z][0]=x;
else ch[z][1]=x;
pushdown(y);
}
inline void zag(int x) {
int y=fa[x],z=fa[y];
fa[y]=x;fa[x]=z;
ch[y][1]=ch[x][0];fa[ch[x][0]]=y;
ch[x][0]=y;
if(y==ch[z][0]) ch[z][0]=x;
else ch[z][1]=x;
pushdown(y);
}
inline void splay(int x,int s) {
while(fa[x]!=s) {
int y=fa[x],z=fa[y];
if(z==s) {
if(x==ch[y][0]) zig(x);
else zag(x);
break;
}
if(y==ch[z][0]) {
if(x==ch[y][0]) zig(y),zig(x);
else zag(x),zig(x);
}
else {
if(x==ch[y][1]) zag(y),zag(x);
else zig(x),zag(x);
}
}
pushdown(x);
if(s==0) root=x;
}
inline void newnode(int &x,int fax,int key) {
x=++tot1;
ch[x][0]=ch[x][1]=0;
fa[x]=fax;
k[x]=key;size[x]=1;
}
/*void debug(int r)
{
if(!r)return;
printf("KEY=%d  NODE=%d  Lchild=%d  Rchild=%d  Size=%d\n",k[r],r,ch[r][0],ch[r][1],size[r]);
debug(ch[r][0]);
debug(ch[r][1]);
} */
inline void ins(int x) {
int p=root,f=0;
while(p) {
if(k[p]>=x) f=p,p=ch[p][0];
else f=p,p=ch[p][1];
}
newnode(p,f,x);
if(k[f]<x) ch[f][1]=p;
else ch[f][0]=p;
splay(p,0);
pushdown(root);
}
inline void inc(int x) {det+=x;}
inline void dec(int x) {
det-=x;
ins(M-det);
res+=size[ch[root][0]];
//cout<<"********"<<res<<endl;
root=ch[root][1];
fa[root]=0;
pushdown(root);
}
int getk(int r,int kx) {
int t=size[ch[r][0]]+1;
if(t==kx) return r;
if (t>kx) return getk(ch[r][0],kx);
return getk(ch[r][1],kx-t);
}
void findx(int kx) {
if(size[root]<kx) {
printf("-1\n");
return;
}
printf("%d\n",k[getk(root,size[root]-kx+1)]+det);
}
int main() {
scanf("%d%d",&n,&M);
char op[5];
int ka;
det=res=0;
while(n--) {
//debug(root);
scanf("%s%d",op,&ka);
if(op[0]=='I'&&ka>=M) ins(ka-det);
else if(op[0]=='A') inc(ka);
else if(op[0]=='S') dec(ka);
else if(op[0]=='F') findx(ka);
}
printf("%d\n",res);
return 0;
}

View Code