bzoj3223 Tyvj 1729 文艺平衡树（Splay Tree+区间翻转）

3223: Tyvj 1729 文艺平衡树

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 2202  Solved: 1226
[Submit][Status][Discuss]

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 Tree处理区间翻转。

       分裂后打标记，然后合并即可。

【代码】

#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#define FOR(a,b,c) for(int a=(b);a<=(c);a++)
using namespace std;

const int maxn = 200000+10;
struct Node{
    Node* ch[2];
    int v,s,flip;
    int cmp(int k) {
        int d=k-ch[0]->s;
        if(d==1) return -1;
        return d<=0? 0:1;
    }
    void maintain() {
        s=ch[0]->s+ch[1]->s+1;
    }
    void pushdown() {
        if(flip) {
            flip=0;
            swap(ch[0],ch[1]);
            ch[0]->flip^=1;
            ch[1]->flip^=1;
        }
    }
};
Node* null=new Node();
void rotate(Node* &o,int d) {
    Node* k=o->ch[d^1]; o->ch[d^1]=k->ch[d]; k->ch[d]=o;
    o->maintain(),k->maintain(); o=k;
}
void splay(Node* &o,int k) {
    o->pushdown();
    int d=o->cmp(k);
    if(d==1) k-=o->ch[0]->s+1;
    if(d!=-1) {
        Node* p=o->ch[d];
        p->pushdown();
        int d2=p->cmp(k);
        int k2=d2==0? k:k-p->ch[0]->s-1;
        if(d2!=-1) {
            splay(p->ch[d2],k2);
            if(d==d2) rotate(o,d^1); else rotate(o->ch[d],d);
        }
        rotate(o,d^1);
    }
}
Node* merge(Node* left,Node* right) {
    splay(left,left->s);
    left->ch[1]=right,left->maintain();
    return left;
}
void split(Node* o,int k,Node* &left,Node* &right) {
    splay(o,k);
    left=o,right=left->ch[1],left->ch[1]=null;
    left->maintain();
}
struct SplaySequence {
    int n;
    Node seq[maxn];
    Node* root;
    
    Node* build(int sz) {
        if(!sz) return null;
        Node* l=build(sz/2);
        Node* o=&seq[++n];
        o->v=n;
        o->ch[0]=l;
        o->ch[1]=build(sz-sz/2-1);
        o->flip=o->s=0;
        o->maintain();
        return o;
    }
    void init(int sz) {
        n=null->s=0;
        root=build(sz);
    }
}spaly;
vector<int> ans;
void print(Node* o) {
    if(o!=null) {
        o->pushdown();
        print(o->ch[0]);
        ans.push_back(o->v);
        print(o->ch[1]);
    }
}

int read() {
    char c=getchar();
    while(!isdigit(c)) c=getchar();
    int x=0;
    while(isdigit(c)) {
        x=x*10+c-'0';
        c=getchar();
    }
    return x;
}
int n,m;
int main() {
    n=read(),m=read();
    spaly.init(n+1);     //在开始添加虚拟结点 
    int l,r;
    Node *left,*right,*mid;
    while(m--) {
        l=read(),r=read();
        split(spaly.root,l,left,right);
        split(right,r-l+1,mid,right);
        mid->flip^=1;
        spaly.root = merge(merge(left,mid),right);
    }
    print(spaly.root);
    for(int i=1;i<ans.size();i++) printf("%d ",ans[i]-1);
    return 0;
}

PS：UVA 11922 Permutation Transformer 简化题目