二逼平衡树（线段树套平衡树）

请你写出一种数据结构，来维护一个长度为 n

的数列，其中需要提供以下操作： 1 l r x，查询整数 x在区间 [l,r]内的排名。
2 l r k，查询区间 [l,r]内排名为 k的值。
3 pos x，将 pos位置的数修改为 x。
4 l r x，查询整数 x在区间 [l,r] 内的前驱(前驱定义为小于 x，且最大的数)。
5 l r x，查询整数 x在区间 [l,r] 内的后继(后继定义为大于 x，且最小的数)。
数列中的位置从左到右依次标号为 1∼n。
区间 [l,r]
表示从位置 l 到位置 r之间（包括两端点）的所有数字。
区间内排名为 k的值指区间内从小到大排在第 k 位的数值。（位次从 1开始）
输入格式
第一行包含两个整数 n,m，表示数列长度以及操作次数。
第二行包含 n个整数，表示有序数列。
接下来 m行，每行包含一个操作指令，格式如题目所述。
输出格式
对于所有操作 1,2,4,5，每个操作输出一个查询结果，每个结果占一行。数据范围
1≤n,m≤5×104,
1≤l≤r≤n,
1≤pos≤n,
1≤k≤r−l+1,
0≤x≤108,
有序数列中的数字始终满足在 [0,108]范围内，
数据保证所有操作一定合法，所有查询一定有解。
输入样例：

9 6
4 2 2 1 9 4 0 1 1
2 1 4 3
3 4 10
2 1 4 3
1 2 5 9
4 3 9 5
5 2 8 5

输出样例：

2
4
3
4
9

思路

我们可以想到树套树， 但要维护一个第k大还带修改的操作， 我们可以想到线段树套平衡树， 而平衡树又选择的是splay
用线段树来维护每个区间， 用平衡树来维护对应区间的数据。
细节见于代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1500010, INF = 1e9;
int n, m;
struct node
{
    int s[2], p, v;
    int size;
    void init(int _v, int _p)
    {
        v = _v, p = _p;
    }
}tr[N];

int L[N], R[N], T[N], idx;
int w[N];

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, z = tr[y].p;
    int k = tr[y].s[1] == x;
    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 &root, int x, int k)
{
    while(tr[x].p != k) // 把k转成x的父节点
    {
        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;
}

void insert(int &root, 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(root, u, 0);
}

void update(int &root, int x, int y)
{
    int u = root;
    while(u)
    {
        if(tr[u].v == x) break;
        if(tr[u].v < x) u = tr[u].s[1];
        else u = tr[u].s[0];
    }
    splay(root, u, 0);
    int l = tr[u].s[0], r = tr[u].s[1];
    while(tr[l].s[1]) l = tr[l].s[1];
    while(tr[r].s[0]) r = tr[r].s[0]; // 查找出来的l, r, u在中序遍历中的位置是 l,u,r
    splay(root, l, 0), splay(root, r, l); // 旋转结果此时r的左儿子中只有u
    tr[r].s[0] = 0;
    pushup(r), pushup(l);
    insert(root, y);
}

void build(int u, int l, int r)
{
    L[u] = l, R[u] = r;
    insert(T[u], -INF), insert(T[u], INF);
    for(int i = l; i <= r; i++) insert(T[u], w[i]);
    if(l == r) return;
    int mid = l + r >> 1;
    build(u<<1, l, mid), build(u<<1|1, mid + 1, r);
}

int get_k(int root, int v) // 查询
{
    int u = root, res = 0;
    while(u)
    {
        if(tr[u].v < v)
            res += tr[tr[u].s[0]].size + 1, u = tr[u].s[1]; //u的左子树节点数加u这个点
        else
            u = tr[u].s[0];
    }
    return res;
}

int query(int u, int a, int b, int x) // 查询a~b区间比x小的数的数目
{
    if(L[u] >= a && R[u] <= b) return get_k(T[u], x) - 1; // 有一个-INF
    int mid = L[u] + R[u] >> 1, res = 0;
    if(a <= mid) res += query(u << 1, a, b, x);
    if(b > mid) res += query(u<<1|1, a, b, x);
    return res;
}

void change(int u, int p, int x)
{
    update(T[u], w[p], x);
    if(L[u] == R[u]) return;
    int mid = L[u] + R[u] >> 1;
    if(p <= mid) change(u <<1, p, x);
    else change(u<<1|1, p, x);
}

int get_pre(int root, int v)
{
    int u = root, res = - INF;
    while(u)
    {
        if(tr[u].v < v) res = max(res, tr[u].v), u = tr[u].s[1];
        else u = tr[u].s[0];
    }
    return res;
}

int get_suc(int root, int v)
{
    int u = root, res = INF;
    while(u)
    {
        if(tr[u].v > v) res = min(res, tr[u].v), u = tr[u].s[0];
        else u = tr[u].s[1];
    }
    return res;
}


int query_pre(int u, int a, int b, int x)
{
    if(L[u] >= a && R[u] <= b) return get_pre(T[u], x);
    int mid = L[u] + R[u] >> 1;
    int res = - INF;
    if(a <= mid) res = max(res, query_pre(u << 1, a, b, x));
    if(b > mid) res = max(res, query_pre(u << 1|1, a, b, x));
    return res;
}

int query_suc(int u, int a, int b, int x)
{
    if(L[u] >= a && R[u] <= b) return get_suc(T[u], x);
    int mid = L[u] + R[u] >> 1, res = INF;
    if(a <= mid) res = min(res, query_suc(u << 1, a, b, x));
    if(b > mid) res = min(res, query_suc(u<<1|1, a, b, x));
    return res;

}
int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++) scanf("%d", &w[i]);
    build(1, 1, n);

    while(m --)
    {
        int op, a, b, x;
        scanf("%d", &op);
        if(op == 1)
        {
            scanf("%d%d%d", &a, &b, &x);
            printf("%d\n", query(1, a, b, x) + 1);
        }
        else if(op == 2)
        {
            scanf("%d%d%d", &a, &b, &x);
            int l = 0, r = 1e8;
            while(l < r)
            {
                int mid = l + r + 1 >> 1;
                if(query(1, a, b, mid) + 1 <= x) l = mid;
                else r = mid - 1;
            }
            printf("%d\n", r);
        }
        else if(op == 3)
        {
            scanf("%d%d", &a, &x);
            change(1, a, x);
            w[a] = x;
        }
        else if(op == 4)
        {
            scanf("%d%d%d", &a, &b, &x);
            printf("%d\n", query_pre(1, a, b, x));
        }
        else
        {
            scanf("%d%d%d", &a, &b, &x);
            printf("%d\n", query_suc(1, a, b, x));
        }
    }
    return 0;
}