P2437 - 【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
因为这里的区间修改是对于整个区间修改的,所以不会影响序列,所以可以用Splay维护序列。
查询k大记一个size即可。然后每次更新值了之后执行删除操作。
记得该下放的时候要下放。
#include<set> #include<map> #include<queue> #include<stack> #include<ctime> #include<cmath> #include<string> #include<vector> #include<cstdio> #include<cstdlib> #include<cstring> #include<iostream> #include<algorithm> #define maxn 100010 #define inf 1999999999 using namespace std; char s[3]; int ch[maxn][2],pre[maxn],size[maxn],key[maxn],lazy[maxn],root=0,tot=0,rt,MIN,ANS=0; void making(int x,int k){ lazy[x]+=k; key[x]+=k; } void updata(int x){ size[x]=size[ch[x][0]]+size[ch[x][1]]+1; } void down(int x){ if(lazy[x]){ if(ch[x][0]) key[ch[x][0]]+=lazy[x],lazy[ch[x][0]]+=lazy[x]; if(ch[x][1]) key[ch[x][1]]+=lazy[x],lazy[ch[x][1]]+=lazy[x]; lazy[x]=0; } } void Rotate(int x,int kind){ int y=pre[x]; down(y),down(x); ch[y][!kind]=ch[x][kind]; pre[ch[x][kind]]=y; if(pre[y]) ch[pre[y]][ch[pre[y]][1]==y]=x; pre[x]=pre[y]; ch[x][kind]=y; pre[y]=x; updata(y),updata(x); } void Splay(int r,int goal){ while(pre[r]!=goal){ if(pre[pre[r]]==goal) Rotate(r,ch[pre[r]][0]==r); else{ int y=pre[r]; int kind=ch[pre[y]][0]==y; if(ch[y][kind]==r){ Rotate(r,!kind); Rotate(r,kind); } else{ Rotate(y,kind); Rotate(r,kind); } } } if(goal==0) root=r; } void newnode(int &x,int fa,int keyy){ x=++tot; pre[x]=fa; ch[x][0]=ch[x][1]; key[x]=keyy; size[x]=1; } void Insert(int k){ down(root); bool f=0; if(ch[root][key[root]<=k]) root=ch[root][key[root]<=k],f=1,Insert(k); if(!f) newnode(ch[root][key[root]<=k],root,k),rt=ch[root][key[root]<=k]; updata(root); } void get_nex(int x,int &ans){ if(!x) return; down(x); if(key[x]<MIN) get_nex(ch[x][1],ans); if(key[x]>=MIN) ans=x,get_nex(ch[x][0],ans); } void Erase(){ int nex1=root;get_nex(root,nex1); Splay(1,0),Splay(nex1,1); ANS+=size[ch[nex1][0]]; ch[nex1][0]=0; updata(nex1); updata(1); } void add(int k){ Splay(1,0); Splay(2,1); making(ch[2][0],k); down(ch[2][0]); Erase(); } void Query(int k){ int rt=root; while(rt){ down(rt); if(size[ch[rt][1]]+1==k) break; if(size[ch[rt][1]]+1<=k){ k-=(size[ch[rt][1]]+1); if(!ch[rt][0]){printf("-1 ");return;} else rt=ch[rt][0]; } else rt=ch[rt][1]; } if(key[rt]==-inf) printf("-1 "); else printf("%d ",key[rt]); } int main() { freopen("!.in","r",stdin); freopen("!.out","w",stdout); int n,x; scanf("%d%d",&n,&MIN); newnode(root,0,-inf); Insert(inf);Splay(rt,0); for(int i=1;i<=n;i++){ scanf("%s%d",s,&x); if(s[0]=='I') if(x>=MIN) Insert(x),Splay(rt,0); if(s[0]=='S') add(-x); if(s[0]=='A') add(x); if(s[0]=='F') Query(x+1); } printf("%d",ANS); return 0; }