hdu3487 Play with Chain splay

题意:

给你一个长度为n1~n序列,要求实现以下两个操作:
CUT i j c:将 ai,ai+1,...aj 切下,接到新数列的第c个数之前。
FLIP i j:将 ai,ai+1,...aj 这段序列翻转。

思路:

用splay维护这段序列。
假设要对区间[l, r]进行操作,则将l-1旋转到根,r+1旋转到根的右子树,则根的右子树左子树就是区间[l, r]
很多操作都基于这个操作。
CUT:先将区间[l, r]旋转到根的右子树的左子树,然后切下来。
要接上去,现将c旋转到根,再把根的右子树的最小值旋转到根的右子树,然后将切下来的区间接到根的右子树的左子树。
FLIP:找到区间[l, r],打上标记即可。

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>

#define MAXN 300005

using namespace std;

int n, q;
int root, sz;
int id[MAXN];
int rev[MAXN];
int siz[MAXN];
int tr[MAXN][2];
int fa[MAXN];

void pushUp(int rt) {
    siz[rt] = siz[tr[rt][0]] + siz[tr[rt][1]] + 1;
}

void pushDown(int rt) {
    if (rev[rt]) {
        int l = tr[rt][0], r = tr[rt][1];
        rev[rt] = 0;
        rev[l] ^= 1; 
        rev[r] ^= 1;
        swap(tr[rt][0], tr[rt][1]);
    }
}

void rotate(int x, int &k) {
    int y = fa[x], z = fa[y];
    int l, r;
    if (tr[y][0] == x) l = 0;
    else l = 1;
    r = l ^ 1;
    if (y == k) k = x;
    else {
        if (tr[z][0] == y) tr[z][0] = x;
        else tr[z][1] = x;
    }
    fa[x] = z;
    fa[y] = x;
    fa[tr[x][r]] = y;
    tr[y][l] = tr[x][r];
    tr[x][r] = y;
    pushUp(y);
    pushUp(x);
}

void splay(int x, int &k) {
    while (x != k) {
        int y = fa[x], z = fa[y];
        if (y != k) {
            if (tr[y][0] == x ^ tr[z][0] == y) rotate(x, k);
            else rotate(y, k);
        }
        rotate(x, k);
    }
}

void build(int l, int r, int rt) {
    if (l > r) return;
    int now = id[l], last = id[rt];
    if (l == r) {
        fa[now] = last;
        siz[now] = 1;
        if (l < rt) tr[last][0] = now;
        else tr[last][1] = now;
        return;
    }
    int mid = l + r >> 1;
    now = id[mid];
    build(l, mid - 1, mid);
    build(mid + 1, r, mid);
    fa[now] = last;
    pushUp(mid);
    if (mid < rt) tr[last][0] = now;
    else tr[last][1] = now;
}

int find(int rt, int rank) {
    pushDown(rt);
    int l = tr[rt][0], r = tr[rt][1];
    if (siz[l] + 1 == rank) return rt;
    if (siz[l] + 1 > rank) return find(l, rank);
    return find(r, rank - siz[l] - 1);
}

int find_min(int rt) {
    pushDown(rt);
    while (tr[rt][0]) {
        rt = tr[rt][0];
        pushDown(rt);
    }
    return rt;
}

void rever(int l, int r) {
    int x = find(root, l);
    int y = find(root, r + 2);
    splay(x, root);
    splay(y, tr[root][1]);
    int z = tr[y][0];
    rev[z] ^= 1;
}

void cut(int l, int r, int c) {
    int x = find(root, l);
    int y = find(root, r + 2);
    splay(x, root);
    splay(y, tr[root][1]);
    int temp = tr[y][0];
    tr[y][0] = 0;
    pushUp(tr[root][1]);
    pushUp(root);
    int z = find(root, c + 1);
    splay(z, root);
    int m = find_min(tr[root][1]);
    splay(m, tr[root][1]);
    tr[tr[root][1]][0] = temp;
    fa[tr[tr[root][1]][0]] = tr[root][1];
    pushUp(tr[root][1]);
    pushUp(root);
}

int main() {
    freopen("3487.in", "r", stdin);
    while (~scanf("%d %d", &n, &q)) {
        if (n == -1 && q == -1) break;
        sz = 0;
        for (int i = 1; i <= n + 2; i++) {
            id[i] = ++sz;
            rev[i] = fa[i] = siz[i] = tr[i][0] = tr[i][1] = 0;
        }
        build(1, n + 2, 0);
        root = n + 3 >> 1;
        char op[8];
        int x, y, z;
        for (int i = 0; i < q; i++) {
            scanf("%s", op);
            if (op[0] == 'C') {
                cin >> x >> y >> z;
                cut(x, y, z);
            } else {
                cin >> x >> y;
                rever(x, y);
            }
        }
        for (int i = 2; i <= n + 1; i++)
            cout << find(root, i) - 1 << " \n"[i == n + 1];
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值