BZOJ3223：Tyvj 1729 文艺平衡树（splay + 区间翻转）

最新推荐文章于 2019-09-03 19:42:00 发布

junior19

最新推荐文章于 2019-09-03 19:42:00 发布

阅读量173

点赞数

分类专栏：平衡树

本文链接：https://blog.csdn.net/junior19/article/details/82428202

版权

平衡树专栏收录该内容

5 篇文章 0 订阅

订阅专栏

Description

您需要写一种数据结构（可参考题目标题），来维护一个有序数列，其中需要提供以下操作：翻转一个区间，例如原有序序列是5 4 3 2 1，翻转区间是[2,4]的话，结果是5 2 3 4 1

Input

第一行为n,m n表示初始序列有n个数，这个序列依次是(1,2……n-1,n) m表示翻转操作次数
接下来m行每行两个数[l,r] 数据保证 1<=l<=r<=n

Output

输出一行n个数字，表示原始序列经过m次变换后的结果

Sample Input

5 3

1 3

1 3

1 4

Sample Output

4 3 2 1 5

HINT

N,M<=100000

思路：splay维护，反转L,R时将L-1转到根部位置，再将R+1转到根的右儿子，然后目标区间就变成一棵子树了，每个节点用个lazy标记是否反转，像线段树一样查询和输出时下传标记就行。

# include <iostream>
# include <cstdio>
# include <cstring>
# include <algorithm>
# include <cstdlib>
using namespace std;
const int maxn = 1e5+30;
const int inf = 2e9+7;
int n;
struct node{
    int val, lazy, siz;
    node *fa, *son[2];
}*root, tree[maxn], *stk[maxn];
int top;
node* newnode(int val, node* fa){//新建节点
    node* t = stk[top--];
    t->val = val;
    t->fa = fa;
    t->lazy = 0;
    t->siz = 1;
    t->son[0] = t->son[1] = NULL;
    return t;
}
void freenode(node* t){//释放节点
    stk[++top] = t;
}
int size(node* t){//返回子树大小
    return t==NULL?0:t->siz;
}
void update(node* t){//更新子树大小
    if(t != NULL){
        t->siz = 1;
        t->siz += size(t->son[0]);
        t->siz += size(t->son[1]);
    }
}
void pushdown(node* t){//下传标记
    if(t->lazy){
        swap(t->son[0], t->son[1]);
        t->lazy = 0;
        if(t->son[0] != NULL) t->son[0]->lazy ^= 1;
        if(t->son[1] != NULL) t->son[1]->lazy ^= 1;
    }
}
bool son(node* f, node* s){//判断s是f的哪个儿子
    if(f == NULL) return 0;
    return f->son[1] == s;
}
void connect(node* f, node* s, bool k){//将s将到f的k儿子上
    if(f != NULL) f->son[k] = s;
    else root = s;
    if(s != NULL) s->fa = f;
}
void rotate(node* s){//将s转到它父亲的位置
    node* f = s->fa;
    node* g = f->fa;
    bool a = son(f, s), b = !a;
    connect(f, s->son[b], a);
    connect(g, s, son(g, f));
    connect(s, f, b);
    update(f);
    update(s);
}
void splay(node* s, node* p){//将s成p的儿子
    if(s == NULL) return;
    while(s->fa != p){
        node* f = s->fa;
        node* g = f->fa;
        if(g == p) rotate(s);
        else{
            if(son(g,f) ^ son(f,s)) rotate(f), rotate(s);
            else rotate(s), rotate(s);
        }
    }
}
node* insert(node* t, int val){//插入一个新的数
    node* res = t->son[val>=t->val];
    if(res == NULL) res = t->son[val>=t->val] = newnode(val, t);
    else res = insert(res, val);
    return res;
}
node* insert(int val){//插入一个数
    if(root == NULL)
        root = newnode(val, NULL);
    else splay(insert(root, val), NULL);
}
node* qry(int th, node* t){//查询排在第th的是哪个点
    pushdown(t);
    if(size(t->son[0]) == th) return t;
    else if(size(t->son[0]) > th) return qry(th, t->son[0]);
    else return qry(th-size(t->son[0])-1, t->son[1]);
}
void print(node* t){//输出
    pushdown(t);
    if(t->son[0] != NULL) print(t->son[0]);
    if(t!=NULL && t->val!=0 && t->val!=n+1) printf("%d ",t->val);
    if(t->son[1] != NULL) print(t->son[1]);
}
int main(){
    int m, l, r;
    scanf("%d%d",&n,&m);
    top = n+10; root = NULL;
    for(int i=1; i<=top; ++i) stk[i] = tree+i;//将节点放进栈里面
    for(int i=0; i<=n+1; ++i) insert(i);
    while(m--){
        scanf("%d%d",&l,&r);
        splay(qry(l-1, root), NULL);
        splay(qry(r+1, root), root);
        root->son[1]->son[0]->lazy ^= 1;
    }
    print(root);
    puts("");
    return 0;
}