4552: [Tjoi2016&Heoi2016]排序

4552: [Tjoi2016&Heoi2016]排序

链接

分析：

　　因为只询问一次，所以考虑二分这个数。显然是没有单调性的，但是我们可以二分所有大于等于mid的数中，是否有满足条件的x（而不是之间判断mid是否满足条件）。

　　那么将大于等于mid的数设为1，小于mid的数设为0，此时对区间排序就变得非常简单了，只需要线段树求一下区间内1的个数，给一段区间赋值即可。

　　最后判断询问的位置是否是1，如果是1，说明这个数是大于等于mid的，否者是小于mid的。

代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<cmath>
#include<cctype>
#include<set>
#include<queue>
#include<vector>
#include<map>
using namespace std;
typedef long long LL;

inline int read() {
    int x=,f=;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-;
    for(;isdigit(ch);ch=getchar())x=x*+ch-'';return x*f;
}

const int N = ;
struct OPT { int opt, a, b; } Q[N];
int a[N], b[N], n, m, k;

struct SegmentTree{
    int sum[N << ], tag[N << ];
    void pushdown(int rt,int len) {
        sum[rt << ] = tag[rt] * (len - (len >> )), tag[rt << ] = tag[rt];
        sum[rt <<  | ] = tag[rt] * (len >> ), tag[rt <<  | ] = tag[rt];
        tag[rt] = -;
    }
    void build(int l,int r,int rt) {
        tag[rt] = -;
        if (l == r) { sum[rt] = b[l]; return ; }
        int mid = (l + r) >> ;
        build(l, mid, rt << ); build(mid + , r, rt <<  | );
        sum[rt] = sum[rt << ]  + sum[rt <<  | ];
    }
    void update(int l,int r,int rt,int L,int R,int v) {
        if (L > R) return ;
        if (L <= l && r <= R) { tag[rt] = v, sum[rt] = v * (r - l + ); return ; }
        if (tag[rt] != -) pushdown(rt, r - l + );
        int mid = (l + r) >> ;
        if (L <= mid) update(l, mid, rt << , L, R, v);
        if (R > mid) update(mid + , r, rt <<  | , L, R, v);
        sum[rt] = sum[rt << ] + sum[rt <<  | ];
    }
    int query(int l,int r,int rt,int L,int R) {
        if (L <= l && r <= R) return sum[rt];
        if (tag[rt] != -) pushdown(rt, r - l + );
        int mid = (l + r) >> , res = ;
        if (L <= mid) res = query(l, mid, rt << , L, R);
        if (R > mid) res += query(mid + , r, rt <<  | , L, R);
        return res;
    }
}T;

bool check(int x) {
    for (int i = ; i <= n; ++i) b[i] = a[i] >= x;
    T.build(, n, );
    for (int i = ; i <= m; ++i) {
        int cnt = T.query(, n, , Q[i].a, Q[i].b);
        if (Q[i].opt) {
            T.update(, n, , Q[i].a, Q[i].a + cnt - , );
            T.update(, n, , Q[i].a + cnt, Q[i].b, );
        } else {
            T.update(, n, , Q[i].a, Q[i].b - cnt, );
            T.update(, n, , Q[i].b - cnt + , Q[i].b, );
        }
    }
    return T.query(, n, , k, k);
}

int main() {
    n = read(), m = read();
    for (int i = ; i <= n; ++i) a[i] = read();
    for (int i = ; i <= m; ++i)
        Q[i].opt = read(), Q[i].a = read(), Q[i].b = read();
    k = read();
    int l = , r = n, ans = ;
    while (l <= r) {
        int mid = (l + r) >> ;
        if (check(mid)) ans = mid, l = mid + ;
        else r = mid - ;
    }
    cout << ans << "\n";
    return ;
}