fhq treap多么强劲。。。
当用无旋treap解决区间/序列问题时,其实每个点所存的值不再对树的形态产生影响,复杂度由随机优先级和堆结构来保证。这棵树不是以点的值为关键字,只是按照操作把一些点进行了修改,使得任何时候对这棵树进行中序遍历,都能得到最新的书架(序列)状态
在节点中,存下书的编号,并且使得每个节点的编号都恰好为他在初始书架中的位置,然后开数组记录书的编号与在初始书架中位置的映射pos[s]。这样对编号s的访问,可以看做在平衡树中对节点编号为pos[s]的操作
也就是说在平衡树中直接能找到对应的数组下标,这样我们就可以对这个序列随便操作了
但是需要一个操作,给出书的编号,我们找他在序列中第几个位置
int find(int pos) {
int siz = tr[tr[pos].ls].siz + 1;
for(; tr[pos].fa; pos = tr[pos].fa)
if(pos == tr[tr[pos].fa].rs)
siz += tr[tr[tr[pos].fa].ls].siz + 1;
return siz;
}
似乎用原来的那些treap操作都难以实现,这个点的位置就是第k大,但是现在没有根据来找第k大,因为不是按点值排序而仅仅是靠位置。。。可以存每个点的父节点,然后这个点若为他父节点的右儿子,那么这个点前面一定有他父节点加他父节点的左儿子点的数量,再加上他自己的左儿子,然后不断往上走,就能确定他的位置(画图理解下就好了)
前面几种操作随便分一下和一下就好了。。。具体看代码
然后对于最后一种操作,只需要对每个点都记录下其代表的编号就好了,因为无旋treap,split操作是给出了两个树根,作为两棵子树的索引,合理的分一下直接输出就好了
实质过程是每次操作一个点,然后把这个点拿出来,按要求放到某个位置,然后询问到的时候由于可以直接访问数组下标,算一算左边有多少个点就能得到这个点目前的位置(size)
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int MAXN = 100000 + 10;
struct trnode{
int val,lz,rnd,ls,rs,siz,s,fa;
}tr[MAXN];
inline void read(int &x) {
x = 0;
int f = 1;
char ch = getchar();
while(ch < '0' || ch > '9') {
if(ch == '-') f = -1;
ch = getchar();
}
while(ch >= '0' && ch <= '9') x = x * 10 + ch - '0', ch = getchar();
x *= f;
}
int n,m,tot,root,pos[MAXN],s[MAXN];
void update(int now) {
tr[now].siz = tr[tr[now].ls].siz + tr[tr[now].rs].siz + 1;
tr[tr[now].ls].fa = now;
tr[tr[now].rs].fa = now;
}
int new_node(int s) {
tr[++tot].s = s;
tr[tot].siz = 1;
tr[tot].rnd = rand();
return tot;
}
int merge(int x, int y) {
if(!x || !y) return x + y;
if(tr[x].rnd < tr[y].rnd) {
tr[x].rs = merge(tr[x].rs, y);
update(x);
return x;
} else {
tr[y].ls = merge(x, tr[y].ls);
update(y);
return y;
}
}
void split(int now, int k, int &x, int &y) {
if(!now) x = y = 0;
else {
if(k <= tr[tr[now].ls].siz)
y = now, split(tr[now].ls, k, x, tr[now].ls);
else
x = now, split(tr[now].rs, k-tr[tr[now].ls].siz-1, tr[now].rs, y);
update(now);
}
}
int find(int pos) {
int siz = tr[tr[pos].ls].siz + 1;
for(; tr[pos].fa; pos = tr[pos].fa)
if(pos == tr[tr[pos].fa].rs)
siz += tr[tr[tr[pos].fa].ls].siz + 1;
return siz;
}
int main() {
scanf("%d %d", &n, &m);
for(int i=1; i<=n; i++) {
read(s[i]);
pos[s[i]] = i;
root = merge(root, new_node(s[i]));
}
for(int i=1; i<=m; i++) {
string cmd;
int s, t, x, y, z, d;
cin >> cmd;
read(s);
int u = find(pos[s]);
if(cmd[0] == 'T') {
split(root, u-1, x, y);
split(y, 1, y, z);
root = merge(y, merge(x, z));
} else if(cmd[0] == 'B') {
split(root, u-1, x, y);
split(y, 1, y, z);
root = merge(x, merge(z, y));
} else if(cmd[0] == 'I') {
read(t);
if(t) {
if(t == 1) {
split(root, u-1, x, y);
split(y, 2, y, z);
split(y, 1, y, d);
root = merge(x, merge(merge(d, y), z));
} else if(t == -1) {
split(root, u-2, x, y);
split(y, 2, y, z);
split(y, 1, y, d);
root = merge(x, merge(merge(d, y), z));
}
}
} else if(cmd[0] == 'A') {
printf("%d\n", u-1);
} else if(cmd[0] == 'Q') {
split(root, s-1, x, y);
split(y, 1, y, z);
printf("%d\n", tr[y].s);
root = merge(x, merge(y, z));
}
}
return 0;
}