题目描述
OIER公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资。如果他心情好,就可能把每位员工的工资加上一个相同的量。反之,如果心情不好,就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。
工资的频繁调整很让员工反感,尤其是集体扣除工资的时候,一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案。
老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第k多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。
好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?
如果某个员工的初始工资低于最低工资标准,那么将不计入最后的答案内
思路
用 splay 维护即可,介绍 splay 的文章可参考 『这里』
#include <cstdio>
const int maxn = 100000 + 10;
int n,m,num,ans,delta,root,father[maxn],ch[maxn][2],key[maxn],cnt[maxn],size[maxn];
inline void clear(int x) { ch[x][0] = ch[x][1] = key[x] = cnt[x] = size[x] = father[x] = 0; }
inline bool getfa(int x) { return ch[father[x]][1] == x; }
inline void Update(int x) {
if (x) {
size[x] = cnt[x];
if (ch[x][0]) size[x] += size[ch[x][0]];
if (ch[x][1]) size[x] += size[ch[x][1]];
}
}
inline void rotate(int x) {
int old = father[x],oldfa = father[father[x]],w = getfa(x);
ch[old][w] = ch[x][w^1]; father[ch[old][w]] = old;
ch[x][w^1] = old; father[old] = x;
father[x] = oldfa;
if (oldfa) ch[oldfa][ch[oldfa][1] == old] = x;
Update(old);
Update(x);
}
inline void splay(int x) {
for (int fa;fa = father[x];rotate(x))
if (father[fa]) rotate(getfa(fa) == getfa(x) ? fa : x);
root = x;
}
inline void Insert(int x) {
if (!root) {
num++;
ch[num][0] = ch[num][1] = father[num] = 0;
key[num] = x;
cnt[num] = size[num] = 1;
root = num;
return;
}
int now = root,fa = 0;
while (true) {
if (key[now] == x) {
cnt[now]++;
Update(now);
Update(fa);
splay(now);
break;
}
fa = now;
now = ch[now][key[now] < x];
if (!now) {
num++;
ch[num][0] = ch[num][1] = 0;
father[num] = fa;
cnt[num] = size[num] = 1;
ch[fa][key[fa] < x] = num;
key[num] = x;
Update(fa);
splay(num);
break;
}
}
}
inline int getnum(int x) {
int now = root,ans = 0;
while (true) {
if (key[now] > x) now = ch[now][0];
else {
if (ch[now][0]) ans += size[ch[now][0]];
if (key[now] == x) {
splay(now);
return ans+1;
}
ans += cnt[now];
now = ch[now][1];
}
}
}
inline int find(int x) {
int now = root;
while (true) {
if (ch[now][0] && x <= size[ch[now][0]]) now = ch[now][0];
else {
int tmp = cnt[now];
if (ch[now][0]) tmp += size[ch[now][0]];
if (x <= tmp) return key[now];
x -= tmp;
now = ch[now][1];
}
}
}
inline void Delete(int x) {
getnum(x);
if (cnt[root] > 1) {
cnt[root]--;
Update(root);
return;
}
if (!ch[root][0] && !ch[root][1]) {
clear(x);
root = 0;
return;
} else if (!ch[root][0] || !ch[root][1]) {
int oldroot = root;
root = ch[root][!ch[root][0]];
father[root] = 0;
clear(oldroot);
return;
}
int l = ch[root][0],oldroot = root;
while (ch[l][1]) l = ch[l][1];
splay(l);
ch[root][1] = ch[oldroot][1];
father[ch[oldroot][1]] = root;
clear(oldroot);
Update(root);
}
int main() {
scanf("%d%d
",&n,&m);
while(n--) {
char op;
int k;
scanf("%s%d",&op,&k);
if (op == 'I' && k >= m) Insert(k-delta);
if (op == 'F') printf("%d
",k <= size[root] ? find(size[root]-k+1)+delta : -1);
if (op == 'A') delta += k;
if (op == 'S') {
int tmp = size[root];
delta -= k;
for (int i = 1;i <= tmp;i++)
if (find(1)+delta < m) {
Delete(find(1));
ans++;
}
}
}
printf("%d",ans);
return 0;
}