# 题解 P1486 【[NOI2004]郁闷的出纳员】

题解 P1486 【[NOI2004]郁闷的出纳员】

题目描述

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

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

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

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

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

输入输出格式

输入格式：

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

接下来的n行，每行表示一条命令。命令可以是以下四种之一：

名称 格式 作用

I命令 I_k 新建一个工资档案，初始工资为k。如果某员工的初始工资低于工资下界，他将立刻离开公司。

A命令 A_k 把每位员工的工资加上k

S命令 S_k 把每位员工的工资扣除k

F命令 F_k 查询第k多的工资

_（下划线）表示一个空格，I命令、A命令、S命令中的k是一个非负整数，F命令中的k是一个正整数。

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

输出格式：

输出文件的行数为F命令的条数加一。

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

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

输入输出样例

输入样例#1：

9 10
I 60
I 70
S 50
F 2
I 30
S 15
A 5
F 1
F 2

输出样例#1：

10
20
-1
2

这道题要求能够支持插入、修改、求第K大的数据结构，首选的数据结构当然是平衡树.

本蒟蒻是用Splay做的，但我的思路似乎和楼下的几位dalao不一样。

对于下列几个操作：

1. I命令 I_k 新建一个工资档案，初始工资为k。如果某员工的初始工资低于工资下界，他将立刻离开公司。

这个操作就是Splay中普通的插入（Insert）

void Insert(int x)
{
    if(x < m) return ;//如果插入的节点的权值小于下界，则不用插入
    int rt = root, p = 0;
    while(t[rt].val != x && rt) p = rt , rt = t[rt].ch[x > t[rt].val];//不停地往下找，p为父亲节点
    if(rt) t[rt].cnt ++;//如果已经有这个数，则副本数＋1
    else//否则新建节点
    {
        rt = ++ tot;
        if(p) t[p].ch[x > t[p].val] = rt;
        t[rt].par = p;
        t[rt].ch[0] = t[rt].ch[1] = 0;
        t[rt].size = t[rt].cnt = 1;
        t[rt].val = x; 
    }
    Splay(rt);
}

2.A命令 A_k 把每位员工的工资加上k

此操作只需要暴力地修改树中每一个节点的权值（val）

void add(int x) {for(int i = 1 ; i <= tot ; i ++) t[i].val += x;}

3.S命令 S_k 把每位员工的工资扣除k

对于这个操作，需要找到第一个扣工资后达到工资下界的人，将他Splay到根节点，删除根节点的左子树即可。最后别忘了减去每个节点的val。（代码中的m为工资下界）

这里还要注意一个细节：有可能扣工资后所有人都要离开公司，所以一开始要插入一个工资为无穷大的节点

    void find(int x)//查找最大的小于等于x的节点并旋转到根
    {
        int rt = root;
        while(x != t[rt].val && t[rt].ch[x > t[rt].val]) rt = t[rt].ch[x > t[rt].val];
        Splay(rt);
    }
    void sub(int x) //扣每个人的工资
    {
        for(int i = 1 ; i <= tot ; i ++) t[i].val -= x;
    }
    int get_next(int x)//查找最小的大于等于x的数
    {
        find(x);
        if(t[root].val >= x) return root;
        int rt = t[root].ch[1];
        while(t[rt].ch[0]) rt = t[rt].ch[0];
        return rt;
    }
   	void Remove(int x)
    {
        int rt = get_next(x + m);
        Splay(rt);
        ans += t[t[rt].ch[0]].size;//根节点的左子树的大小即为删除的人数
        t[rt].ch[0] = 0;//删除左子树
        pushup(rt);//更新rt的大小
        sub(x);
    }

4.F命令 F_k 查询第k多的工资

普通的查询操作，注意因为先前插入的无穷大节点，所以要查询第k+1大的节点

int Kth(int k)
{
    if(k >= t[root].size) return -1;
    k++;//因为先前插入的无穷大节点，所以要查询第k+1大的节点
    int rt = root;
    while(1)
    {
        if(t[rt].ch[1] && k <= t[t[rt].ch[1]].size) rt = t[rt].ch[1];
        else if(k > t[t[rt].ch[1]].size + t[rt].cnt)
        {
            k = k - t[t[rt].ch[1]].size - t[rt].cnt;
            rt = t[rt].ch[0];
        }
        else return t[rt].val;
    }
}

最后附上AC代码

开O2 : 302ms

不开O2：570ms

你可以在最优解第8页中找到我

#include<bits/stdc++.h>
using namespace std;
const int N = 1000009 , INF = 1547483647;//INF不能太大，否则爆int
struct Splay{int ch[2] , par , cnt , size , val;} t[N];
/*ch[2]分别表示左右儿子
  par表示父亲
  cnt为副本数
  size为树的大小
  val为节点权值（每个人的工资）
*/
int n  , m , tot = 0 , root , ans = 0 ;
//m为工资下界，root为根节点编号，ans为离开公司的人数
int read()//快读不解释
{
    int s = 0 , w = 1;char ch = getchar();
    while(ch <'0' || ch > '9'){if(ch == '-') w = -1; ch = getchar();}
    while(ch >= '0' && ch <= '9'){s = (s << 1) + (s << 3) + (ch ^ 48);ch = getchar();}
    return s * w;
} 
char read_char()
{
    char ch = getchar();
    while(ch != 'I' && ch != 'S' && ch != 'F' && ch != 'A') ch = getchar();
    return ch;
}
bool chk(int rt){return t[t[rt].par].ch[1] == rt;} //判断rt是父亲的那个儿子
void pushup(int rt){ t[rt].size = t[t[rt].ch[0]].size + t[t[rt].ch[1]].size + t[rt].cnt;}
//更新操作
void rotate(int rt)//旋转操作
{
    int f = t[rt].par , gf = t[f].par , d = chk(rt) , son = t[rt].ch[d ^ 1];
    t[f].ch[d] = son; t[son].par = f;
    t[gf].ch[chk(f)] = rt; t[rt].par = gf;
    t[rt].ch[d ^ 1] = f; t[f].par = rt;
    pushup(f) , pushup(rt); 
}
void Splay(int rt , int goal = 0)//将rt旋转到指定节点的儿子，若调用时不输入goal，则默认为0（即旋转到根节点）
{
    while(t[rt].par != goal)
    {
        int f = t[rt].par ;int  gf = t[f].par;
        if(gf != goal)
        {
            if(chk(rt) == chk(f)) rotate(f);
            else rotate(rt);
        }
        rotate(rt);
    }
    if(!goal) root = rt;
}
void Insert(int x)//插入
{
    if(x < m) return ;
    int rt = root, p = 0;
    while(t[rt].val != x && rt) p = rt , rt = t[rt].ch[x > t[rt].val];
    if(rt) t[rt].cnt ++;
    else
    {
        rt = ++ tot;
        if(p) t[p].ch[x > t[p].val] = rt;
        t[rt].par = p;
        t[rt].ch[0] = t[rt].ch[1] = 0;
        t[rt].size = t[rt].cnt = 1;
        t[rt].val = x; 
    }
    Splay(rt);
}
void find(int x)//查找最大的小于等于x的节点并旋转到根
{
    int rt = root;
    while(x != t[rt].val && t[rt].ch[x > t[rt].val]) rt = t[rt].ch[x > t[rt].val];
    Splay(rt);
}
void add(int x) {for(int i = 1 ; i <= tot ; i ++) t[i].val += x;}//＋每个人的工资

void sub(int x) {for(int i = 1 ; i <= tot ; i ++) t[i].val -= x;} //扣每个人的工资

int get_next(int x)//查找最小的大于等于x的数
{
    find(x);
    if(t[root].val >= x) return root;
    int rt = t[root].ch[1];
    while(t[rt].ch[0]) rt = t[rt].ch[0];
    return rt;
}
void Remove(int x)//让工资低于下界的人离开
{
    int rt = get_next(x + m);
    Splay(rt);
    ans += t[t[rt].ch[0]].size;
    t[rt].ch[0] = 0;
    pushup(rt);
    sub(x);
}

int Kth(int k)//求第k大
{
    if(k >= t[root].size) return -1; k++;
    int rt = root;
    while(1)
    {
        if(t[rt].ch[1] && k <= t[t[rt].ch[1]].size) rt = t[rt].ch[1];
        else if(k > t[t[rt].ch[1]].size + t[rt].cnt)
        {
            k = k - t[t[rt].ch[1]].size - t[rt].cnt;
            rt = t[rt].ch[0];
        }
        else return t[rt].val;
    }
}

int main()
{
    n = read(); m = read();Insert(INF);
    while(n --)
    {
        char opt = read_char();
        int x = read();
        if(opt == 'I') Insert(x);
        if(opt == 'A') add(x);
        if(opt == 'S') Remove(x);
        if(opt == 'F') printf("%d\n",Kth(x));
    }
    cout << ans << endl;
    return 0;

}