【题目描述】
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。
输出文件的最后一行包含一个整数,为离开公司的员工的总数。
S a m p l e I n p u t Sample~~Input Sample Input
9 10 I 60 I 70 S 50 F 2 I 30 S 15 A 5 F 1 F 2
S a m p l e O u t p u t Sample~~Output Sample Output
10 20 -1 2
【题意分析】
I操作直接插入x,注意低于初始工资直接return,不然挂 A操作暴力全部节点权值都加上x S操作的话,先找到权值为x+lim(即调整后正好要退出公司的那个节点)节点的后缀,将它旋到根,那么它的左子树全部是要离开的,注意maintain F操作直接Kth,因为插了哨兵所以是Kth(x+1)
Code:
#include <iostream> #include <cstdio> #include <cstdlib> #include <cstring> #include <cctype> #include <algorithm> #define MAXN 300000 #define INF 1 << 29 using namespace std; int n, lim, ans, sz; struct _splay { int son[MAXN][3], size[MAXN], father[MAXN], cnt[MAXN], val[MAXN], root; inline void maintain (int x) { size[x] = size[son[x][0]] + size[son[x][1]] + cnt[x]; } inline void rotate (int x) { int y = father[x], z = father[y]; int k = son[y][1] == x, kk = son[z][1] == y; son[z][kk] = x; father[x] = z; son[y][k] = son[x][k ^ 1]; father[son[x][k ^ 1]] = y; son[x][k ^ 1] = y; father[y] = x; maintain (y), maintain (x); } inline void splay (int x, int goal) { while (father[x] != goal) { int y = father[x], z = father[y]; if (z != goal) (son[y][1] == x) ^ (son[z][1] == y) ? rotate (x) : rotate (y); rotate (x); } if (! goal) root = x; } inline void find (int x) { int now = root; while (son[now][x > val[now]] && x != val[now]) now = son[now][x > val[now]]; splay (now, 0); } inline void addition (int x) { for (register int i = 1; i <= sz; i++) val[i] += x; } inline void subtraction (int x) { for (register int i = 1; i <= sz; i++) val[i] -= x; } inline int succ (int x) { int now = root, nxt; while (now) { (val[now] >= x)//查询第一个大于等于的 ? nxt = now, now = son[now][0] : now = son[now][1]; } return nxt; } inline void remove (int x) { int low = succ (lim + x); splay (low, 0), ans += size[son[low][0]]; son[low][0] = 0, maintain (low); subtraction (x); } int Kth (int x, int rank) {//查询的是第rank大 if (son[x][1] && rank <= size[son[x][1]]) return Kth (son[x][1], rank); return (rank <= size[son[x][1]] + cnt[x]) ? x : Kth (son[x][0], rank - size[son[x][1]] - cnt[x]); } inline void insert (int x) { if (x < lim) return;//初始工资低于下限不插入 int now = root, fa = 0; while (val[now] != x && now) fa = now, now = son[now][x > val[now]]; if (now) cnt[now]++; else { now = ++sz; if (fa) son[fa][x > val[fa]] = now; size[now] = cnt[now] = 1; son[now][0] = son[now][1] = 0; val[now] = x, father[now] = fa; } splay (now, 0); } }tree; int main () { scanf ("%d%d", &n, &lim), tree.insert (INF), tree.insert (-INF); for (register int i = 1; i <= n; i++) { char opt[5]; int x; scanf ("%s%d", opt, &x); if (opt[0] == 'I') tree.insert (x); if (opt[0] == 'A') tree.addition (x); if (opt[0] == 'S') tree.remove (x); if (opt[0] == 'F') { if (x >= tree.size[tree.root]) puts ("-1");//如果x比总结点个数还要多,输出-1 else printf ("%d\n", tree.val[tree.Kth (tree.root, x + 1)]); } } printf ("%d\n", ans); return 0; }