BZOJ 3196: Tyvj 1730 二逼平衡树 窒息的操作

传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=3196
题意：
1.查询k在区间内的排名
2.查询区间内排名为k的值
3.修改某一位值上的数值
4.查询k在区间内的前驱(前驱定义为小于x，且最大的数)
5.查询k在区间内的后继(后继定义为大于x，且最小的数)
思路：窒息的操作
code：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int mod = 1e9 + 7;
const int maxn = 5e4 + 1000;
inline void read(int &x){x = 0;char p = getchar();while(!(p <= '9' && p >= '0'))p = getchar();while(p <= '9' && p >= '0')x *= 10, x += p - 48, p = getchar();}
int d[maxn], m, n, p, s, t, v;
typedef vector<int>vec;
vec a[maxn];
inline void ins(vec&a, int v){ a.insert(lower_bound(a.begin(), a.end(), v), v);}
inline void del(vec&a, int v){ a.erase(lower_bound(a.begin(), a.end(), v));}
inline void ins(int i, int v){ for(; i <= n; i += i & -i) ins(a[i], v);}
inline void del(int i, int v){ for(; i <= n; i += i & -i)del(a[i], v);}
inline int ask(vec&a, int v) {return lower_bound(a.begin(), a.end(), v) - a.begin();}
inline int ask(int v, int s, int t)
{
    int k = 1;
    for(; s; s ^= s & -s)k -= ask(a[s], v);
    for(; t; t ^= t & -t)k += ask(a[t], v);
    return k;
}
inline int cal(int v, int s, int t)
{
    int l = 0, r = 1e8;
    while(l ^ r)
    {
        int j = l + r + 1 >> 1;
        v >= ask(j, s, t) ? l = j : r = j - 1;
    }
    return l;
}
int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++i) read(d[i]);
    for(int i = 1; i <= n; ++i) ins(i, d[i]);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d", &p, &s, &t);
        if(p == 3)
        {
            del(s, d[s]), ins(s, d[s] = t);///修改某一位置上的数值；
            continue;
        }
        int ans=0;
        --s, scanf("%d", &v);
        if(p == 1) ans=ask(v, s, t);///查询 x在区间内的排名；
        if(p == 2) ans=cal(v, s, t);///查询区间内排名为 k 的值；
         ///查询 x x x 在区间内的前趋（前趋定义为小于 x，且最大的数）；
        if(p == 5) ans=cal(ask(v + 1, s, t), s, t);
        ///查询 x x x 在区间内的后继（后继定义为大于 x，且最小的数）。
        printf("%d\n",ans);
    }
    return 0;
}