P2596 [ZJOI2006]书架 - 无旋treap

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;
}
posted @ 2018-10-18 14:49  Zolrk  阅读(198)  评论(0编辑  收藏  举报