题目:
http://www.lydsy.com/JudgeOnline/problem.php?id=1503
题意:
Description
OIER公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资。如果他心情好,就可能把每位员工的工资加上一个相同的量。反之,如果心情不好,就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。工资的频繁调整很让员工反感,尤其是集体扣除工资的时候,一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案。老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第k多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?
Input
Output
输出文件的行数为F命令的条数加一。对于每条F命令,你的程序要输出一行,仅包含一个整数,为当前工资第k多的员工所拿的工资数,如果k大于目前员工的数目,则输出-1。输出文件的最后一行包含一个整数,为离开公司的员工的总数。
思路:
智障了一下午,还以为我伸展树哪里写挫了,,,用工资作为值,全部员工加减工资用lazy操作,删除小于最低工资的员工先用二分确定要删除的区间,然后删除,找第k大工资,因为是按树的中序升序建树,那么实际就是找第(siz[root]-2-a+1)大(减2的原因是因为插入了两个边界节点)
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define key_val son[son[root][1]][0]
const int N = 110010, INF = 0x3f3f3f3f;
int son[N][2], pre[N], key[N], val[N], siz[N], lazy[N];
int n, num, root, cnt;
void new_node(int &x, int fa, int v)
{
x = ++num;
pre[x] = fa, key[x] = v;
siz[x] = 1, lazy[x] = 0;
son[x][0] = son[x][1] = 0;
}
void push_up(int x)
{
siz[x] = siz[son[x][0]] + siz[son[x][1]] + 1;
}
void push_down(int x)
{
if(lazy[x])
{
key[x] += lazy[x];
lazy[son[x][0]] += lazy[x], lazy[son[x][1]] += lazy[x];
lazy[x] = 0;
}
}
void _rotate(int x, int dir)
{
int y = pre[x];
push_down(y), push_down(x);
son[y][!dir] = son[x][dir], pre[son[x][dir]] = y;
if(pre[y]) son[pre[y]][son[pre[y]][1]==y] = x;
pre[x] = pre[y];
son[x][dir] = y, pre[y] = x;
push_up(y);
}
void splay(int x, int goal)
{
push_down(x);
while(pre[x] != goal)
{
int y = pre[x], z = pre[y];
push_down(z), push_down(y), push_down(x);
if(pre[y] == goal) _rotate(x, son[y][0] == x);
else
{
int dir = son[pre[y]][0] == y;
if(son[y][dir] == x) _rotate(x, !dir), _rotate(x, dir);
else _rotate(y, dir), _rotate(x, dir);
}
}
push_up(x);
if(goal == 0) root = x;
}
int get_kth(int k)
{
int x = root;
push_down(x);
while(siz[son[x][0]] + 1 != k)
{
if(siz[son[x][0]] + 1 > k) x = son[x][0];
else k -= (siz[son[x][0]] + 1), x = son[x][1];
push_down(x);
}
return x;
}
void _delete(int l, int r)
{
splay(get_kth(l-1), 0);
splay(get_kth(r+1), root);
key_val = 0;
push_up(son[root][1]), push_up(root);
}
void _insert(int v)
{
if(! root) new_node(root, 0, v);
else
{
int x = root;
push_down(x);
while(son[x][key[x]<v])
{
x = son[x][key[x]<v];
push_down(x);
}
new_node(son[x][key[x]<v], x, v);
splay(son[x][key[x]<v], 0);
}
}
void add(int v)
{
splay(get_kth(1), 0);
splay(get_kth(siz[root]), root);
lazy[key_val] += v;
}
int main()
{
int minn, a, cnt = 0;
char ch;
scanf("%d%d", &n, &minn);
root = num = 0;
_insert(-INF), _insert(INF);
for(int i = 1; i <= n; i++)
{
scanf(" %c%d", &ch, &a);
if(ch == 'I')
{
if(a < minn) continue;
else _insert(a);
}
else if(ch == 'A') add(a);
else if(ch == 'S')
{
add(-a);
int l = 2, r = siz[root], res = -1;
while(l <= r)
{
int mid = (l + r) >> 1;
if(key[get_kth(mid)] < minn) l = mid + 1, res = mid;
else r = mid - 1;
}
if(res != -1)
{
cnt += res - 2 + 1;
_delete(2, res);
}
}
else
{
if(a > siz[root] - 2) printf("-1\n");
else
{
a = siz[root] - 2 - a + 1;
printf("%d\n", key[get_kth(a+1)]);
}
}
}
printf("%d\n", cnt);
return 0;
}