【BZOJ1503】【NOI2004】郁闷的出纳员

【问题描述】

    OIER公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把他们的工资扣除一个相同的量。 工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。 老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第k多的员工拿多少工资。 请你编一个工资统计程序。

输入：

【分析】

    平衡树。我写splay。

    工资的加减，由于是对于所有员工的，所以不用修改每个员工的值，只要把总共的变化值delt记下来就行了。加入员工的时候，把初始工资减去delt。查询时第k多的工资，把工资下界减去delt。

    查询第k多的工资，也很简单。只要在平衡树每个节点多记一个参数，表示以它为根的子树的节点数。把第k多转换成第K小。然后从根开始遍历，如果当前点的左儿子子树节点数等于K-1的话，就找到了。如果小于，那么往左走。反之往右走。往右走记得要把K扣除比它小的数的数目，即当前点左儿子子树节点数+1.

【代码】

    写得一向龊的splay。。。

/**************************************************************
Problem: 1503
User: N_C_Derek
Language: C++
Result: Accepted
Time:760 ms
Memory:5180 kb
****************************************************************/

//noi2004郁闷的出纳员
#include <cstring>
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
int n,m,Min,delt,root,t,ans;
struct node
{
int l,r,num,father,data;
}tree[200001];
void zig(int x)
{
int p = tree[x].father;
tree[x].father = tree
.father;
if (tree[p].father != 0)
{
if (tree[tree[p].father].l == p)
tree[tree[p].father].l = x;
else tree[tree[p].father].r = x;
}
tree[p].father = x;
tree[tree[x].r].father = p;
tree[p].l = tree[x].r;
tree[x].r = p;
tree[p].num -= tree[tree[x].l].num + 1;
tree[x].num += tree[tree[p].r].num + 1;
}
void zag(int x)
{
int p = tree[x].father;
tree[x].father = tree[p].father;
if (tree[p].father != 0)
{
if (tree[tree[p].father].l == p)
tree[tree[p].father].l = x;
else tree[tree[p].father].r = x;
}
tree[p].father = x;
tree[tree[x].l].father = p;
tree[p].r = tree[x].l;
tree[x].l = p;
tree[p].num -= tree[tree[x].r].num + 1;
tree[x].num += tree[tree[p].l].num + 1;
}
void splay(int x)
{
while (tree[x].father)
{
int j = tree[x].father;
if (tree[j].father == 0)
{
if (tree[j].l == x) zig(x);else zag(x);
break;
}
if (tree[tree[j].father].l == j)
{
if (tree[j].l == x)
{
zig(j);
zig(x);
}
else {
zag(x);zig(x);
}
}else
{
if (tree[j].r == x)
{
zag(j);
zag(x);
}
else {
zig(x);zag(x);
}
}
}
root = x;
}
void Insert(int x)
{
if (!root) {tree[++t].data = x;tree[t].num = 1;root = t;return;}
int i = root,j;
while (i)
{
j = i;
tree[i].num ++;
if (x < tree[i].data)
i = tree[i].l;
else i = tree[i].r;
}
if (x < tree[j].data)
tree[j].l = ++t;
else
tree[j].r = ++t;
tree[t].father = j;
tree[t].data = x;
tree[t].num = 1;
splay(t);
}
void find(int k)
{
if (k < 1)
{
printf("-1\n");
return;
}
int i = root;
while (tree[i].l || tree[i].r)
{
if (k == tree[tree[i].l].num + 1) break;
if (k < tree[tree[i].l].num + 1)
i = tree[i].l;
else
{
k -= tree[tree[i].l].num + 1;
i = tree[i].r;
}
}
printf("%d\n",tree[i].data + delt);
}
void Delete(int k)
{
int x = root;
int xx= 0;
while (x)
{
if (tree[x].data < k) {xx = x;x = tree[x].r;}
else x = tree[x].l;
}
if (xx == 0) return;
splay(xx);
if (tree[xx].r == 0) {ans += tree[xx].num;root = 0;return;}
ans += tree[tree[xx].l].num + 1;
root = tree[xx].r;tree[root].father = 0;
}
int main()
{
scanf("%d %d",&m,&Min);
for (int i = 1;i <= m;i ++)
{
char ch;
int k;
scanf("\n");
scanf("%c %d",&ch,&k);
if (ch == 'I')
{
if (k >= Min) Insert(k - delt);
}
else if (ch == 'F')
{
find(tree[root].num - k + 1);
}
else if (ch == 'S')
{
delt -= k;
Delete(Min - delt);
}
else
delt += k;
}
printf("%d",ans);
}