洛谷 P1486 [NOI2004] 郁闷的出纳员（splay）

编程拾光

于 2024-07-08 16:46:10 发布

阅读量500

点赞数 8

CC 4.0 BY-SA版权

分类专栏：算法题文章标签：算法数据结构

本文链接：https://blog.csdn.net/weixin_74174531/article/details/140272673

算法题专栏收录该内容

9 篇文章

订阅专栏

题目描述

OIER 公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把当前在公司的所有员工的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。

工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。

老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第 𝑘k 多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。

好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难吧？

如果某个员工的初始工资低于最低工资标准，那么将不计入最后的答案内。

输入格式

第一行有两个整数 𝑛 和 min⁡。𝑛n 表示下面有多少条命令，min⁡ 表示工资下界。

接下来的 𝑛 行，每行一个字符 𝑥 和一个整数 𝑘，表示一条命令。命令可以是以下四种之一：

I k 新建一个工资档案，初始工资为 𝑘。如果某员工的初始工资低于工资下界，他将立刻离开公司。
A k 把每位员工的工资加上 𝑘。
S k 把每位员工的工资扣除 𝑘。
F k 查询第 𝑘 多的工资。

在初始时，可以认为公司里一个员工也没有。

输出格式

对于每条 F 命令，你的程序要输出一行，仅包含一个整数，为当前工资第 𝑘 多的员工所拿的工资数，如果 𝑘大于目前员工的数目，则输出 −1。

输出的最后一行包含一个整数，为离开公司的员工的总数。

请注意，初始工资低于工资下界的员工不算做离开公司的员工。

输入输出样例

输入 #1复制

输出 #1复制

说明/提示

数据规模与约定

对于全部的测试点，保证：

I 命令的条数不超过 1e5；
A 和 S 命令的总条数不超过 100；
F 命令的条数不超过 1e5；
每次工资调整的调整量不超过 1e3；
新员工的工资不超过 1e5。
0≤𝑛≤3×1e5，0≤min≤1e9，输入的所有数字均在 32 位带符号整形范围内。

本题有以下几种情况：

(1)招一个员工

(2)给所有员工同时涨薪(+k)

(3)给所有员工同时降薪(-k)

(4)在剩余员工中找到工资第k大数

对于2,3两种情况，我们设置一个变量delta，每次涨薪，降薪的时候我们都在deltag上操作，那么每个节点的当前价值加上delta就是实际价值，每次降薪之后，我们需要找到>=min-delta的最小值的工资的节点，然后把这个节点之前的所有节点都删除（不包括L节点）。也就是每次删除的时候都是删除前缀。

与上一个题的最大的区别就是本题每个节点的价值就是节点的价值，上一个题每个节点的价值就是下标。

代码如下：

#include<bits/stdc++.h>
using namespace std;
const int N=100010,INF=1e9;
int n,m,delta;

struct Node
{
    int s[2],p,v;
    int size;
    
    void init(int _v,int _p)
    {
        v=_v;
        p=_p;
        size=1;
    }
}tr[N];
int root,idx;

void pushup(int x)
{
    tr[x].size=tr[tr[x].s[0]].size+tr[tr[x].s[1]].size+1;
}

void rotate(int x)//左旋右旋操作
{
    int y=tr[x].p;//y是x的父亲
    int z=tr[y].p;//z是y的父亲
    int k=tr[y].s[1]==x;//k==1是x是y的右儿子，否则x是y的左儿子
    
    tr[z].s[tr[z].s[1]==y]=x;
    tr[x].p=z;
    tr[y].s[k]=tr[x].s[k^1];
    tr[tr[x].s[k^1]].p=y;
    tr[x].s[k^1]=y;
    tr[y].p=x;
    
    pushup(y);
    pushup(x);
}

void splay(int x,int k)
{
    while(tr[x].p!=k)
    {
        int y=tr[x].p,z=tr[y].p;
        if(z!=k)
            if((tr[y].s[1]==x)^(tr[z].s[1]==y))//折线
                rotate(x);
            else
                rotate(y);
        rotate(x);
    }
    if(!k)
        root=x;
}

int insert(int v)
{
    int u=root,p=0;
    while(u)
    {
        p=u;
        u=tr[u].s[v>tr[u].v];
    }
    u=++idx;
    if(p)
        tr[p].s[v>tr[p].v]=u;
        tr[u].init(v,p);
        splay(u,0);

    return u;
}

int get(int v)//大于等于某一个数的最小的节点编号
{
    int u=root,res;
    while(u)
    {
        if(tr[u].v>=v)
        {
            res=u;
            u=tr[u].s[0];
        }
        else
            u=tr[u].s[1];
    }
    return res;
}

int get_k(int k)//返回第k小的数
{
    int u=root;
    while(u)
    {
        if(tr[tr[u].s[0]].size>=k)
            u=tr[u].s[0];
        else if(tr[tr[u].s[0]].size+1==k)
            return tr[u].v;
        else 
        {
            k-=tr[tr[u].s[0]].size+1;
            u=tr[u].s[1];
        }
    }
    return -1;
}

int main()
{
    scanf("%d%d",&n,&m);
    int L=insert(-INF),R=insert(INF);//左右端点
    
    int tot=0;//表示总人数
    while(n--)
    {
        char op[2];
        int k;
        scanf("%s%d",op,&k);
        if(op[0]=='I')//来了一个新员工
        {
            if(k>=m)
            {
                k-=delta;//k+delta表示真实工资，所以这里k减去delta
                insert(k);
                tot++;
            }
        }
        else if(op[0]=='A')//所有人工资加上k
        delta+=k;
        else if(op[0]=='S')//所有人工资减去k
        {
            delta-=k;
            R=get(m-delta);//找到大于等于m-delta的最小的节点编号
            splay(R,0);
            splay(L,R);
            tr[L].s[1]=0;//删除（将L的右子树清空）
            
            pushup(L);
            pushup(R);
        }
        else//输出第k多的工资
        {
            if(tr[root].size-2<k)//k>当前员工数量
                puts("-1");
            else//第k+1大的数是第tr[root].size-k小的数（包含哨兵节点）
                printf("%d\n",get_k(tr[root].size-k)+delta);
        }
        
    }
    printf("%d\n",tot-(tr[root].size-2));//离开公司的员工总数
    
    
    return 0;
}