BZOJ4552: [Tjoi2016&Heoi2016]排序(线段树 二分)

题意

题目链接

sol

神仙二分orz

首先二分一个答案，表示假设询问的位置为$x$，把$>= x$的看成$1$，$< x$的看成$0$

对形成的$0/1$序列模拟，线段树维护。

最后根据询问位置上的数缩小答案范围

这题的单调性不好想，但是证起来其实也挺显然的，考虑如果答案$x$最后对应的值为$1$，那么$<x$的值最后对应的答案也一定为$1$

#include<bits/stdc++.h>
#define pair pair<int, int> 
#define mp(x, y) make_pair(x, y)
#define fi first
#define se second
using namespace std;
const int maxn = 4e6 + 10, inf = 1e9 + 10;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-')f =- 1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}
int n, m, q;
int a[maxn], b[maxn], opt[maxn], l[maxn], r[maxn];
#define ls k << 1
#define rs k << 1 | 1
struct node {
    int l, r, siz, tag, cnt[2];
}t[maxn];   
pair operator + (const pair &a, const pair &b) {
    return mp(a.fi + b.fi, a.se + b.se);
}
void update(int k) {
    for(int i = 0; i <= 1; i++) t[k].cnt[i] = t[ls].cnt[i] + t[rs].cnt[i];
}
void ps(int k, int val) {
    t[k].cnt[0] = t[k].cnt[1] = 0;
    t[k].cnt[val] = t[k].siz;
    t[k].tag = val;
}
void pushdown(int k) {
    if(t[k].tag == -1) return ;
    ps(ls, t[k].tag); ps(rs, t[k].tag);
    t[k].tag = -1;
}
void build(int k, int ll, int rr) {
    t[k].l = ll; t[k].r = rr; t[k].siz = rr - ll + 1; t[k].tag = -1; t[k].cnt[0] = t[k].cnt[1] = 0;
    if(ll == rr) {t[k].cnt[b[ll]]++; return ;}
    int mid = ll + rr >> 1;
    build(ls, ll, mid); build(rs, mid + 1, rr);
    update(k);
}
pair query(int k, int ll, int rr) {
    if(ll <= t[k].l && t[k].r <= rr) return mp(t[k].cnt[0], t[k].cnt[1]);
    int mid = t[k].l + t[k].r >> 1;
    pushdown(k);
    if(ll > mid) return query(rs, ll, rr);
    if(rr <= mid) return query(ls, ll, rr);
    return query(ls, ll, rr) + query(rs, ll, rr);
}
void mem(int k, int ll, int rr, int val) {
    if(ll <= t[k].l && t[k].r <= rr) {
        ps(k, val); return ;
    }
    pushdown(k);
    int mid = t[k].l + t[k].r >> 1;
    if(ll <= mid) mem(ls, ll, rr, val);
    if(rr >  mid) mem(rs, ll, rr, val);
    update(k);
}
int point(int k, int pos) {
    if(t[k].l == t[k].r) {
        if(t[k].cnt[1]) return 1;
        else return 0;
    }
    pushdown(k);
    int mid = t[k].l + t[k].r >> 1;
    if(pos <= mid) return point(ls, pos);
    else return point(rs, pos);
}
void dfs(int k) {
    if(t[k].l == t[k].r) {
        if(t[k].cnt[1]) printf("1 ");
        else printf("0 ");
        return ;
    }
    pushdown(k);
    dfs(ls); dfs(rs);
}
bool check(int val) {
    for(int i = 1; i <= n; i++) b[i] = (a[i] >= val);
    build(1, 1, n);
    //dfs(1); puts("");
    for(int i = 1; i <= m; i++) {
        pair now = query(1, l[i], r[i]);
        if(opt[i] == 0) mem(1, l[i], l[i] + now.fi - 1, 0), mem(1, l[i] + now.fi, r[i], 1);
        else mem(1, l[i], l[i] + now.se - 1, 1), mem(1, l[i] + now.se, r[i], 0);
    //  dfs(1); puts("");
    }
    return point(1, q);
}
main() {
    n = read(); m = read();
    for(int i = 1; i <= n; i++) a[i] = read();
    for(int i = 1; i <= m; i++) opt[i] = read(), l[i] = read(), r[i] = read();
    q = read();
    int l = 1, r = n, ans = -1;
    while(l <= r) {
        int mid = l + r >> 1;
        if(check(mid)) ans = mid, l = mid + 1;
        else r = mid - 1;
    }
    printf("%d", ans);
}