题目来源:[HEOI2016/TJOI2016] 排序 - 洛谷https://www.luogu.com.cn/problem/P2824
问题思路
本文介绍一种二分答案的做法,时间复杂度为:(n+m)*log(n)*log(n).本题存在nlog(n)的做法,然而其做法没有二分答案的做法通俗易懂.
默认读者已读过原题,故直接介绍思路.
- 二分枚举答案为mid,对于原序列中大于或等于mid的元素,将其转化为1,对于其他元素,将其转化为0.
- 现在,原排列已转化为01序列,m次操作对01序列进行排序.
- 经过m次排序操作后,如果第p个位置上的元素为1,那么更新二分边界 r = mid.否则如果第p个位置上的元素为0,那么更新二分边界 l = mid.
- 如何维护m次操作呢?该问题转化为线段树的修改+查找问题。没有接触过线段树修查问题的读者,建议先去学习相关知识,此处不多赘述.
好了,思路介绍完了,你可能惊讶于为什么二分答案可行.如下便是二分答案可行性的一些解释.
假如经过m次操作后,第p个位置上的元素为x0。我们考虑m次普通的序列排序,但是此次排序咱们让每一个元素都夹带“私货”.对应于问题思路中的第一步,令mid = x0,对于x>=mid的元素,让它带上额外的元素1.对于x<x0的元素,让它带上额外的元素0.经过m次操作后,元素x0位于第p个位置上,且额外的“私货”——元素1,也随之到达了第p个位置.
看到这里是否嗅到了二分答案的味道?既然 mid = x0可行,那么按照mid = x0 - 1,x0 - 2....的要求,给元素分配额外“私货”0和1,那么 1 依旧会随着元素x0来到第p个位置上.然而,假如mid>x0,那么跟随元素x0到达第p个位置上的额外“私货”便是元素0了.
经过上述解释,你应该理解了二分答案在该题目中的可行性.
当我们不再考虑普通的序列排序,而是考虑01序列的排序时,对元素0和1进行排序后的结果,必定能够映射到其普通序列的正确的排序结果上.举个例子,对序列 perm = [1,5,2,4,3] 进行排序,先二分答案 mid = 3,那么perm 转化 01序列 bperm = [0,1,0,1,1],那么对 bperm进行升序排序的结果为: bperm = [0,0,1,1,1],该排序结果能够映射到其普通序列的正确排序结果上,即
==> perm = [1,2,3,4,5].
OK,解释到此结束,更多细节问题请参考代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 100005;
int a[N];
int ls(int u) { return u << 1; }
int rs(int u) { return u << 1 | 1; }
struct tree {
int l, r, s, tag;
}tr[N * 4];
void build(int u, int l, int r) {
tr[u] = { l,r,0,-1 };
if (l == r)
return;
int mid = l + r >> 1;
build(ls(u), l, mid);
build(rs(u), mid + 1, r);
}
void addTag(int u, int v) {
tr[u].s = v * (tr[u].r - tr[u].l + 1);
tr[u].tag = v;
}
void pu(int u) {
tr[u].s = tr[ls(u)].s + tr[rs(u)].s;
}
void pd(int u) {
// tag == -1代表该线段的标签已重置...
if (tr[u].tag != -1) {
addTag(ls(u), tr[u].tag);
addTag(rs(u), tr[u].tag);
tr[u].tag = -1;
}
}
void update(int L, int R, int u, int v) {
if (tr[u].l > R || tr[u].r < L) return;
if (L <= tr[u].l && tr[u].r <= R) {
addTag(u, v);
return;
}
pd(u);
int mid = (tr[u].l + tr[u].r) >> 1;
if (L <= mid) update(L, R, ls(u), v);
if (R > mid) update(L, R, rs(u), v);
pu(u);
}
int query(int L, int R, int u) {
if (tr[u].l > R || tr[u].r < L) return 0;
if (L <= tr[u].l && tr[u].r <= R) return tr[u].s;
pd(u);
int mid = (tr[u].l + tr[u].r) >> 1;
int ans = 0;
if (L <= mid) ans = query(L, R, ls(u));
if (R > mid) ans = ans + query(L, R, rs(u));
return ans;
}
int main() {
ios::sync_with_stdio(0), cin.tie(0);
int n, m;
cin >> n >> m;
for (int i = 1;i <= n;i++) cin >> a[i];
vector<array<int, 3>>op(m);
for (int i = 0;i < m;i++) {
cin >> op[i][0] >> op[i][1] >> op[i][2];
}
build(1, 1, n);
int q;
cin >> q;
int l = 0, r = n + 1;
while (l + 1 != r) {
update(1, n, 1, 0);
int mid = l + r >> 1;
for (int i = 1;i <= n;i++)
if (a[i] >= mid) update(i, i, 1, 1);
for (int i = 0;i < m;i++) {
int o = op[i][0], ql = op[i][1], qr = op[i][2];
int k = query(ql, qr, 1);
int cnt = qr - ql + 1;
// [qr-k+1,qr]全为1,[ql,qr-k]全为0...
if (k == 0 || k == cnt) continue;
if (o == 0) {
update(ql, qr - k, 1, 0);
update(qr - k + 1, qr, 1, 1);
}
else {
update(ql, ql + k - 1, 1, 1);
update(ql + k, qr, 1, 0);
}
}
query(q, q, 1) == 1 ? l = mid : r = mid;
}
cout << l << '\n';
return 0;
}