【洛谷】P1486 郁闷的出纳员

题目地址：

https://www.luogu.com.cn/problem/P1486

题目描述：
OIER 公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把当前在公司的所有员工的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第 $k$ 多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难吧？如果某个员工的初始工资低于最低工资标准，那么将不计入最后的答案内。

输入格式：
第一行有两个整数 $n$ 和 $\min$。$n$ 表示下面有多少条命令，$\min$ 表示工资下界。

接下来的 $n$ 行，每行一个字符 $x$ 和一个整数 $k$，表示一条命令。命令可以是以下四种之一：
I k 新建一个工资档案，初始工资为 $k$。如果某员工的初始工资低于工资下界，他将立刻离开公司。
A k 把每位员工的工资加上 $k$。
S k 把每位员工的工资扣除 $k$。
F k 查询第 $k$ 多的工资。
在初始时，可以认为公司里一个员工也没有。

输出格式：
对于每条 F 命令，你的程序要输出一行，仅包含一个整数，为当前工资第 $k$ 多的员工所拿的工资数，如果 $k$ 大于目前员工的数目，则输出 $-1$。
输出的最后一行包含一个整数，为离开公司的员工的总数。
请注意，初始工资低于工资下界的员工不算做离开公司的员工。

数据范围：
对于全部的测试点，保证：
I 命令的条数不超过 $10^5$；
A 和 S 命令的总条数不超过 $100$；
F 命令的条数不超过 $10^5$；
每次工资调整的调整量不超过 $10^3$；
新员工的工资不超过 $10^5$。
$0\le n\le 3\times 10^5$，$0\le \text{min}\le 10^9$，输入的所有数字均在 $32$ 位带符号整形范围内。

其实就是设计一个数据结构，可以做如下操作：
1、插入一个数；
2、将所有数加上或减去一个数；
3、将所有小于等于某个数的节点全删掉；
4、查第$k$大的数。
可以用FHQ Treap做。操作1、3、4是可以直接支持的，操作2可以用一个全局变量来做，该全局变量是所有节点值的一个偏移量，即所有节点的真实值是其值加上偏移量，这样就不需要FHQ Treap的节点存储真实节点信息了。操作2就直接操作偏移量本身即可。当然如果是减去一个数，则还需要查一下真实值小于最小工资的节点，将这些节点全删掉。而这些节点在中序遍历里一定是连续的，所以直接分裂，然后将左半边整个删除即可。代码如下：

#include <iostream>
using namespace std;

namespace {
const int N = 4e5 + 10;
int n, m;
struct Node {
#define lc(p) tr[p].l
#define rc(p) tr[p].r
#define key(p) tr[p].key
#define sz(p) tr[p].sz
  int l, r;
  int key, val;
  int sz;
} tr[N];
int idx, root;
// 全局偏移量
int delta;
int x, y, z;
// 记录一下离职的员工数
int res;

#define pushup(p) sz(p) = sz(lc(p)) + sz(rc(p)) + 1

int get_node(int k) {
  tr[++idx] = {0, 0, k, rand(), 1};
  return idx;
}

int merge(int x, int y) {
  if (!x || !y) return x | y;
  if (tr[x].val > tr[y].val) {
    rc(x) = merge(rc(x), y);
    pushup(x);
    return x;
  } else {
    lc(y) = merge(x, lc(y));
    pushup(y);
    return y;
  }
}

// 按key分裂成小于等于k和大于k的两部分分别赋值给x和y
void split(int p, int k, int &x, int &y) {
  if (!p) x = y = 0;
  else {
    if (k >= key(p)) x = p, split(rc(p), k, rc(p), y);
    else y = p, split(lc(p), k, x, lc(p));
    pushup(p);
  }
}

void insert(int k) {
  split(root, k, x, z);
  y = get_node(k);
  root = merge(merge(x, y), z);
}

void minus_k() {
  // 节点x要被删掉当且仅当x + delta < m，即x + delta + 1 <= m
  split(root, m - delta - 1, x, y);
  // 累加离职人数
  res += sz(x);
  root = y;
}

int get_k(int k) {
  int u = root;
  while (u) {
    if (k <= sz(lc(u))) u = lc(u);
    else if (sz(lc(u)) + 1 < k)
      k -= sz(lc(u)) + 1, u = rc(u);
    else return key(u);
  }

  return -1;
}
}  // namespace

int main() {
  scanf("%d%d", &n, &m);
  while (n--) {
    char op[2];
    int k;
    scanf("%s%d\n", op, &k);
    if (op[0] == 'I') {
      if (k < m) continue;
      insert(k - delta);
      // 如果是涨工资，那肯定不会有人离职，直接改偏移量即可
    } else if (op[0] == 'A') delta += k;
    // 如果是减工资，则有可能有员工离职，需要做查询
    else if (op[0] == 'S') {
      delta -= k;
      minus_k();
    } else {
      if (k > sz(root)) puts("-1");
      else
        printf("%d\n", get_k(sz(root) - k + 1) + delta);
    }
  }

  printf("%d\n", res);
}

操作1、3、4时间复杂度$O(\log n)$，操作$2$如果是加上一个数，则时间$O(1)$，否则$O(\log n)$，空间$O(n)$。