[NOI2004][BZOJ1503] 郁闷的出纳员
1503: [NOI2004]郁闷的出纳员
Time Limit: 5 Sec Memory Limit: 64 MBSubmit: 7749 Solved: 2725
[Submit][Status][Discuss]
Description
OIER公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第k多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难吧？
Input
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
I 60
I 70
S 50
F 2
I 30
S 15
A 5
F 1
F 2
Sample Output
10
20
-1
2
20
-1
2
HINT
I命令的条数不超过100000 A命令和S命令的总条数不超过100 F命令的条数不超过100000 每次工资调整的调整量不超过1000 新员工的工资不超过100000
Source
treap 800ms...裸题
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<algorithm> #include<cmath> #include<queue> using namespace std; char c[1]; int n,mn,x,cnt,delta,root,leave; int size[100005],num[100005],l[100005],r[100005],rnd[100005]; void updata(int x) { size[x]=size[l[x]]+size[r[x]]+1; } void rturn(int &k) { int t=l[k];l[k]=r[t];r[t]=k;updata(k);updata(t);k=t; } void lturn(int &k) { int t=r[k];r[k]=l[t];l[t]=k;updata(k);updata(t);k=t; } void insert(int &k,int x) { if (!k) { k=++cnt; num[k]=x; rnd[k]=rand(); size[k]=1; return; } size[k]++; if (x<num[k]) { insert(l[k],x); if (rnd[l[k]]<rnd[k]) rturn(k); } else { insert(r[k],x); if (rnd[r[k]]<rnd[k]) lturn(k); } } int del(int &k,int x) { int t; if (!k) return 0; if (num[k]<x) { t=size[l[k]]+1; k=r[k]; return t+del(k,x); } else { t=del(l[k],x); size[k]-=t; return t; } } int find(int k,int x) { if (size[l[k]]+1==x) return num[k]+delta; if (size[l[k]]+1>x) return find(l[k],x); if (size[l[k]]+1<x) return find(r[k],x-size[l[k]]-1); } int main() { scanf("%d%d",&n,&mn); while(n--) { scanf("%s%d",c,&x); if (c[0]=='I') if (x>=mn) insert(root,x-delta); if (c[0]=='A') delta+=x; if (c[0]=='S') { delta-=x; leave+=del(root,mn-delta); } if (c[0]=='F') if (x>size[root]) printf("-1\n"); else printf("%d\n",find(root,size[root]-x+1)); } printf("%d",leave); return 0; }
splay 1300ms
#include<iostream> #include<cstdio> #include<cstring> #include<cstdlib> #include<algorithm> #include<cmath> #include<set> using namespace std; char c[5]; int n,x,mn,root,cnt,delta,leave,tree[100001][2],size[100001],fa[100001],num[100001]; inline void update(int k) { size[k]=size[tree[k][0]]+size[tree[k][1]]+1; } inline void rotate(int x,int &k) { int y=fa[x],z=fa[y],l,r; l=(tree[y][1]==x);r=l^1; if (y==k) k=x; else tree[z][tree[z][1]==y]=x; fa[x]=z; fa[y]=x; fa[tree[x][r]]=y; tree[y][l]=tree[x][r]; tree[x][r]=y; update(y); update(x); } inline void splay(int x,int &k) { while (x!=k) { int y=fa[x],z=fa[y]; if (y!=k) { if ((tree[y][0]==x)^(tree[z][0]==y)) rotate(x,k); else rotate(y,k); } rotate(x,k); } } void insert(int k) { if(!root) { root=++cnt; num[cnt]=k; size[cnt]=1; return; } int p=root,z; while(p) { z=p; ++size[p]; if(k<num[p])p=tree[p][0]; else p=tree[p][1]; } if (num[z]>k) tree[z][0]=++cnt; else tree[z][1]=++cnt; num[cnt]=k;size[cnt]=1;fa[cnt]=z; splay(cnt,root); } int del(int &x,int f) { if(!x)return 0; int k; if(num[x]+delta<mn) { k=del(tree[x][1],x)+size[tree[x][0]]+1; size[tree[x][1]]=size[x]-k; x=tree[x][1];fa[x]=f; } else { k=del(tree[x][0],x); size[x]-=k; } return k; } int find(int k,int x) { if (size[tree[k][1]]>=x) return find(tree[k][1],x); if (size[tree[k][1]]+1==x) return num[k]; return find(tree[k][0],x-size[tree[k][1]]-1); } int main() { scanf("%d%d",&n,&mn); for (int i=1;i<=n;i++) { scanf("%s%d",c,&x); if (c[0]=='I'&&x>=mn) insert(x-delta); if (c[0]=='A') delta+=x; if (c[0]=='S') { delta-=x; leave+=del(root,0); } if (c[0]=='F') printf("%d\n",x<=size[root]?find(root,x)+delta:-1); } printf("%d",leave); return 0; }
