[HDU 5649] DZY Loves Sorting （线段树+二分）

链接

HDU 5649

---

题意

给出一个1到n的全排列，然后m组操作，对每组操作(opt, l, r)，若opt为0，则将区间[l, r]内元素升序排序，若opt为1，则倒序排序，最终给出一个k，问m组操作后，第k个元素的值。

---

题解

是一道很棒的二分题，注意到题目条件的特殊性，一个在于“全排列”，一个在于最终只给出了一个需要查询的位置。
想象下m组排序过后序列的样子，如果我们将大于等于b的元素看做1，小于b的元素看做0，在b从1增长到n的过程中，这个序列中1的个数是递增的，能用二分的原因就在于此，即“单调性”。
我们可以对b二分，然后把排序前的序列中的元素大于等于b的看做1否则0，显然m组排序过后的序列和先将序列排序后再分1和0得到的序列是相同的。这样原本升降序的操作就可以用线段树的区间查询和成段更新来完成，最终的复杂度是nlognlogn。

---

代码

#include <cstdio>
#include <iostream>
using namespace std;
#define maxn (100010)
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
int seg[maxn << 2], lazy[maxn << 2], a[maxn];
void build(int value, int l, int r, int rt)
{
    lazy[rt] = -1;
    if(l == r)
    {
        seg[rt] = (a[l] >= value) ? 1 : 0;
        return ;
    }
    int m = (l + r) >> 1;
    build(value, lson), build(value, rson);
    seg[rt] = seg[rt<<1] + seg[rt<<1|1];
}
inline void _pushdown(int rt, int l, int r)
{
    if(lazy[rt] == -1) return ;
    lazy[rt<<1] = lazy[rt<<1|1] = lazy[rt];
    int m = (l + r) >> 1;
    seg[rt<<1] = lazy[rt<<1] ? m - l + 1 : 0;
    seg[rt<<1|1] = lazy[rt<<1|1] ? r - m : 0;
    lazy[rt] = -1;
}
int query(int L, int R, int l, int r, int rt)
{
    if(L <= l && r <= R) { return seg[rt]; }
    _pushdown(rt, l, r);
    int m = (l + r) >> 1, ret = 0;
    if(L <= m) ret += query(L, R, lson);
    if(R > m) ret += query(L, R, rson);
    return ret;
}
void update(int L, int R, int value, int l, int r, int rt)
{
    if(L <= l && r <= R)
    {
        lazy[rt] = value;
        seg[rt] = value ? r - l + 1 : 0;
        return ;
    }
    _pushdown(rt, l, r);
    int m = (l + r) >> 1;
    if(L <= m) update(L, R, value, lson);
    if(R > m) update(L, R, value, rson);
    seg[rt] = seg[rt<<1] + seg[rt<<1|1];
}
struct _Q
{
    int opt, l, r;
} Q[maxn];
int main()
{
    int t, n, m, k, o;
    cin >> t;
    while(t--)
    {
        cin >> n >> m;
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]);
        for(int i = 0; i < m; i++)
            scanf("%d%d%d", &Q[i].opt, &Q[i].l, &Q[i].r);
        cin >> k;
        o = 0;
        int l = 1, r = n, mid;
        while(l <= r)
        {

            mid = (l + r) >> 1;
            //printf("l = %d, r = %d, mid = %d\n", l, r, mid);
            build(mid, 1, n, 1);
            /*
            for(int i = 1; i <= n; i++)
            {
                printf("[%d] ", query(i, i, 1, n, 1));
            } printf("\n");
            */
            for(int i = 0, ret, x, y; i < m; i++)
            {
                ret = query(Q[i].l, Q[i].r, 1, n, 1);
                if(!ret || ret == Q[i].r - Q[i].l + 1) continue;
                //printf("ret = %d\n", ret);
                if(Q[i].opt)
                {
                    update(Q[i].l, Q[i].l + ret - 1, 1, 1, n, 1);
                    update(Q[i].l + ret, Q[i].r, 0, 1, n, 1);
                }
                else
                {
                    update(Q[i].l, Q[i].r - ret, 0, 1, n, 1);
                    update(Q[i].r - ret + 1, Q[i].r, 1, 1, n, 1);
                }
                /*
                for(int i = 1; i <= n; i++)
                {
                    printf("[%d] ", query(i, i, 1, n, 1));
                } printf("\n");
                */
            }
            //cout << "b[k] = " << query(k, k, 1, n, 1) << endl;
            if(query(k, k, 1, n, 1)) { o = mid; l = mid + 1; }
            else r = mid - 1;
        }
        cout << o << endl;
    }
    return 0;
}