[luogu3380][bzoj3196]【模板】二逼平衡树【树套树】

题目地址

【洛谷传送门】

题目大意

区间查询k的排名，查找k排名的数，单点修改，区间前驱，区间后继。

感想

真的第一次写树套树，整个人都不对了。重构代码2次，发现样例都过不了，splay直接爆炸，可能是我太弱了。
换了treap，就过掉了。
但是理解树套树的思路也花了我不少的时间。

分析

很多人都不知道树套树是什么东西，我一开始也是很懵逼的。
为什么两棵树可以套在一起？？这是什么操作？？
但是HG的xyc大佬给我点波了一句话，我就明白了。
因为原来的线段树中只有tag一个标记，我们记录的东西就太少了。也不方便操作，那么我们就将这个线段树的所有节点都换成一棵平衡树就可以了。
如果有时间我一定会写一个树套树的详细总结（flag）。
但是我们不是对每一个节点都建一个关于整体的平衡树的节点，这样复杂度太高了，我们就将这个区间内的所有点都扔到一个平衡树中，是不是非常暴力呢？？
其实这样的做法，仅仅增加了空间复杂度\(log_2n \times n\)的空间，因为我们需要建多棵平衡树，所以我们一开始就准备一大堆的空节点，如果有需要的，那么就直接从空节点中拿一个就可以了，
那么关于时间复杂度，其实也就只是增加了一个平衡树的单次操作的\(log\)。
但是有一个例外，就是操作2。
因为我们需要查找排名k的数，那么我们选择二分答案，每一次枚举rank，然后检验是否在k内，那么操作2还需要一个log的复杂度，所以差不多就是\(log_n^3 \times n\)的复杂度。

代码

#include <bits/stdc++.h>
#define inf 2147483647
#define lc (nod << 1)
#define rc (nod << 1 | 1)
#define N 500004
using namespace std;
template <typename T>
inline void read(T &x) {x = 0; T fl = 1;char ch = 0;while (ch < '0' || ch > '9') {if (ch == '-') fl = -1;ch = getchar();}while (ch >= '0' && ch <= '9') {x = (x << 1) + (x << 3) + (ch ^ 48);ch = getchar();}x *= fl;
}
struct node {int val, cnt, ch[2], sz, rd;void init(int x) {val = x; cnt = sz = 1; ch[1] = ch[0] = 0; rd = rand() % 100;}
}tr[N * 20];
int tot;
struct Treap {#define ls tr[nod].ch[0]#define rs tr[nod].ch[1]void pushup(int nod) {tr[nod].sz = tr[ls].sz + tr[rs].sz + tr[nod].cnt;}void rotate(int &nod, int d) {int k = tr[nod].ch[d];tr[nod].ch[d] = tr[k].ch[d ^ 1];tr[k].ch[d ^ 1] = nod;pushup(nod); pushup(k);nod = k;}void ins(int &nod, int val) {if (!nod) {tr[nod = ++ tot].init(val);return;}tr[nod].sz ++;if (tr[nod].val == val) {tr[nod].cnt ++;return;}int d = val > tr[nod].val;ins(tr[nod].ch[d], val);if (tr[nod].rd > tr[tr[nod].ch[d]].rd) rotate(nod, d);}void del(int &nod, int val) {if (!nod) return;if (tr[nod].val == val) {if (tr[nod].cnt > 1) {tr[nod].cnt --;tr[nod].sz --;return;}int d = tr[ls].rd > tr[rs].rd;if (ls == 0 || rs == 0) nod = ls + rs;else rotate(nod, d), del(nod, val);}else tr[nod].sz --, del(tr[nod].ch[tr[nod].val < val], val);}int rk(int nod, int val) {if (!nod) return 0;if (tr[nod].val == val) return tr[ls].sz;if (tr[nod].val > val) return rk(ls, val);else return tr[ls].sz + tr[nod].cnt + rk(rs, val);}int kth(int nod, int k)  {while (233) {if (k <= tr[ls].sz) nod = ls;else if (k > tr[ls].sz + tr[nod].cnt) k -= tr[ls].sz + tr[nod].cnt, nod = rs;else return tr[nod].val;}}int pre(int nod, int val) {if (!nod) return -inf;if (tr[nod].val >= val) return pre(ls, val);else return max(tr[nod].val, pre(rs, val));}int suc(int nod, int val) {if (!nod) return inf;if (tr[nod].val <= val) return suc(rs, val);else return min(tr[nod].val, suc(ls, val));}
}tp[N << 2];
int n, m;
int a[N], rt[N];
void build(int nod, int l, int r) {for (int i = l; i <= r; i ++) tp[nod].ins(rt[nod], a[i]);if (l == r) return;int mid = (l + r) >> 1;build(lc, l, mid);build(rc, mid + 1, r);
}
int query_rk(int nod, int l, int r, int ql, int qr, int k) {if (l > qr || r < ql) return 0;if (ql <= l && r <= qr) return tp[nod].rk(rt[nod], k);int mid = (l + r) >> 1, res = 0;res += query_rk(lc, l, mid, ql, qr, k);res += query_rk(rc, mid + 1, r, ql, qr, k);return res;
}
void update_point(int nod, int l, int r, int k, int val) {if (k < l || k > r) return;tp[nod].del(rt[nod], a[k]);tp[nod].ins(rt[nod], val);if (l == r) return;int mid = (l + r) >> 1;update_point(lc, l, mid, k, val);update_point(rc, mid + 1, r, k, val);
}
int query_pre(int nod, int l, int r, int ql, int qr, int k) {if (l > qr || r < ql) return -inf;if (ql <= l && r <= qr) return tp[nod].pre(rt[nod], k);int mid = (l + r) >> 1, res = query_pre(lc, l, mid, ql, qr, k);res = max(res, query_pre(rc, mid + 1, r, ql, qr, k));return res;
}
int query_suc(int nod, int l, int r, int ql, int qr, int k) {if (l > qr || r < ql) return inf;if (ql <= l && r <= qr) return tp[nod].suc(rt[nod], k);int mid = (l + r) >> 1, res = query_suc(lc, l, mid, ql, qr, k);res = min(res, query_suc(rc, mid + 1, r, ql, qr, k));return res;
}
int query_fd(int ql, int qr, int k) {int l = 0, r = 1e8, res = -1;while (l <= r) {int mid = (l + r) >> 1;if (query_rk(1, 1, n, ql, qr, mid) + 1 <= k) res = mid, l = mid + 1;else r = mid - 1;}return res;
}
int main() {srand(time(NULL));tot = 0;read(n); read(m);for (int i = 1; i <= n; i ++) read(a[i]);build(1, 1, n);for (int i = 1; i <= m; i ++) {int opt, x, y, z;read(opt);if (opt == 1) {read(x); read(y); read(z);printf("%d\n", query_rk(1, 1, n, x, y, z) + 1); }if (opt == 2) {read(x); read(y); read(z);printf("%d\n", query_fd(x, y, z));}if (opt == 3) {read(x); read(y);update_point(1, 1, n, x, y);a[x] = y;}if (opt == 4) {read(x); read(y); read(z);int res = query_pre(1, 1, n, x, y, z);printf("%d\n", res);}if (opt == 5) {read(x); read(y); read(z);int res = query_suc(1, 1, n, x, y, z);printf("%d\n", res);}}return 0;
}