HYSBZ 1503 郁闷的出纳员 splay

题目:

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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值