排序[HEOI2016/TJOI2016]

【题目描述】
在 \(2016\) 年,佳媛姐姐喜欢上了数字序列。因而她经常研究关于序列的一些奇奇怪怪的问题,现在她在研究一个难题,需要你来帮助她。

这个难题是这样子的:给出一个 \(1\) 到 \(n\) 的排列,现在对这个排列序列进行 \(m\) 次局部排序,排序分为两种:

  • 0 l r 表示将区间 \([l,r]\) 的数字升序排序
  • 1 l r 表示将区间 \([l,r]\) 的数字降序排序

注意,这里是对下标在区间 \([l,r]\) 内的数排序。
最后询问第 \(q\) 位置上的数字。

【输入格式】
输入数据的第一行为两个整数 \(n\) 和 \(m\),\(n\) 表示序列的长度,\(m\) 表示局部排序的次数。

第二行为 \(n\) 个整数,表示 \(1\) 到 \(n\) 的一个排列。

接下来输入 \(m\) 行,每一行有三个整数 \(\text{op},l,r\),\(\text{op}\) 为 \(0\) 代表升序排序,\(\text{op}\) 为 \(1\) 代表降序排序, \(l,r\) 表示排序的区间。

最后输入一个整数 \(q\),表示排序完之后询问的位置。

【输出格式】
输出数据仅有一行,一个整数,表示按照顺序将全部的部分排序结束后第 \(q\) 位置上的数字。

\(n,m≤10_5\) ,\(1\leq q\leq n\)

题解

发现\(m\)非常大 用一般的排序肯定会超时

有什么东西处理区间操作比较快呢 当然是线段树啦

但是线段树并不支持区间排序

所以我们把问题转换一下:

如果要我们用线段树排序一个只由0和1组成的序列 显然是可以通过区间求和以及区间修改做到的

我们可以二分答案\(mid\) 然后把输入排列中所有大于等于\(mid\)的置为1 其余的置为0

然后用线段树来进行这\(m\)次对于01序列的排序

排好后 如果位置\(q\)上的数字为1 则表示\(mid\)大于等于答案 否则则是\(mid\)小于答案

时间复杂度\(O(n \log n + m \log^2 n)\)

代码

#include <bits/stdc++.h>
#define lson ind<<1
#define rson ind<<1|1
using namespace std;

inline int read() {
    int x = 0, f = 1; char ch = getchar();
    for (; ch > '9' || ch < '0'; ch = getchar()) if (ch == '-') f = -1;
    for (; ch <= '9' && ch >= '0'; ch = getchar()) x = (x << 1) + (x << 3) + (ch ^ '0');
    return x * f;
}

int n, m, k, a[100005], tp[100005], le[100005], ri[100005], tmp[100005]; 

namespace Segtree{
    struct segtree{
        int l, r, sum, tag;
    } tr[400005];
    
    void build(int ind, int l, int r) {
        tr[ind].l = l; tr[ind].r = r; tr[ind].tag = -1;
        if (l == r) {
            tr[ind].sum = tmp[l]; return;
        }
        int mid = (l + r) >> 1;
        build(lson, l, mid); build(rson, mid+1, r);
        tr[ind].sum = tr[lson].sum + tr[rson].sum;
    }
    
    inline void pushdown(int ind) {
        if (tr[ind].tag == -1) return; 
        int v = tr[ind].tag; tr[ind].tag = -1;
        tr[lson].tag = v; tr[lson].sum = (tr[lson].r - tr[lson].l + 1) * v;
        tr[rson].tag = v; tr[rson].sum = (tr[rson].r - tr[rson].l + 1) * v;
    }
    
    void update(int ind, int x, int y, int v) {
        if (x > y) return;
        int l = tr[ind].l, r = tr[ind].r;
        if (x <= l && r <= y) {
            tr[ind].tag = v; tr[ind].sum = (r - l + 1) * v; return;
        }
        int mid = (l + r) >> 1;
        pushdown(ind);
        if (x <= mid) update(lson, x, y, v);
        if (mid < y) update(rson, x, y, v);
        tr[ind].sum = tr[lson].sum + tr[rson].sum;
    } 
    
    int query(int ind, int x, int y) {
        int l = tr[ind].l, r = tr[ind].r;
        if (x <= l && r <= y) {
            return tr[ind].sum;
        }
        int mid = (l + r) >> 1, ret = 0;;
        pushdown(ind);
        if (x <= mid) ret += query(lson, x, y);
        if (mid < y) ret += query(rson, x, y);
        return ret;
    }
}

using namespace Segtree;

bool check(int x) {
    for (int i = 1; i <= n; i++) tmp[i] = (a[i] >= x);
    build(1, 1, n); 
    for (int i = 1; i <= m; i++) {
        int num = query(1, le[i], ri[i]);
        if (!tp[i]) {
            update(1, le[i], ri[i]-num, 0); update(1, ri[i] - num + 1, ri[i], 1);  
        } else {
            update(1, le[i], le[i] + num - 1, 1); update(1, le[i] + num, ri[i], 0);  
        }
    }
    return query(1, k, k) == 1;
}

int main() {
    n = read(); m = read();
    for (int i = 1; i <= n; i++) a[i] = read();
    for (int i = 1; i <= m; i++) tp[i] = read(), le[i] = read(), ri[i] = read();
    k = read();
    int l = 1, r = n, mid, ans = 0;
    while (l <= r) {
        mid = (l + r) >> 1;
        if (check(mid)) {
            ans = mid; l = mid + 1;
        } else r = mid - 1;
    }
    printf("%d\n", ans);
    return 0;
}
上一篇:2020牛客寒假算法基础集训营4


下一篇:1009 说反话 (20分)