【splay】splay小結 文藝平衡樹

題目:文藝平衡樹

如果不懂基礎知識可以看本蒟蒻寫的

treap小結

本題是動態的序列翻轉
首先我們知道splay是可以分裂和合並的
我們可以打一個lazy標記,表示這棵子樹有沒有翻轉過
一開始不知道如何建樹
想過Insert
但是效率低並且有點不像splay
所以就來了一個Build
要注意先序遍歷
後來調試發現rotate寫反了(捂臉)

注意Splay中的

Node* w=o->ch[d];
if(d==e)Rotate(o,d^1);
else Rotate(o->ch[d],e^1);

不能寫成

Node* w=o->ch[d];
if(d==e)Rotate(o,d^1);
else Rotate(w,e^1);

因爲這裏是一段引用

rotate中二段旋轉不能寫錯,否則效率降低

Print時要pushdown

分裂時先把p-1旋轉到根節點
然後取右子樹
然後把q旋轉到根節點
取左子樹就是【p,q】了

有一個一個虛擬節點0
保證split不出錯

完整ac代碼

#include<cstdio>
#include<iostream>
#define N 100010
using namespace std;
int n,num,p,q,m;
struct Node{
    Node* ch[2];
    int v,s,f;
    int cmp(int x){
        if(ch[0]->s+1==x)return -1;
        return ch[0]->s+1 < x;
    }
    void maintain(){
        s=ch[0]->s+ch[1]->s+1;
    }
    void pushdown(){
        if(f){
            f=0;
            swap(ch[0],ch[1]);
            ch[0]->f=!ch[0]->f;
            ch[1]->f=!ch[1]->f;
        }
    }
};
Node* null=new Node();
Node *ro,*l,*r,*mid,*o;
void Build(Node* &o,int sz){
    if(!sz){
        o=null;
        return ;
    }
    o=new Node();
    Build(o->ch[0],sz/2);
    o->v=num++;
    Build(o->ch[1],sz-sz/2-1);
    o->maintain();
}
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* w=o->ch[d];
        w->pushdown();
        int e= w->cmp(k);
        if(e==1) k-= w->ch[0]->s+1;
        if(e!=-1){
            Splay(w->ch[e],k);
            if(d==e)Rotate(o,d^1);
            else Rotate(o->ch[d],e^1);
        }
        Rotate(o,d^1);
    }
}
Node* Merge(Node* l,Node* r){
    Splay(l,l->s);
    l->ch[1]=r;
    l->maintain();
    return l;
}
void Split(Node* &o,int k,Node* &l,Node* &r){
    Splay(o,k);
    l=o;
    r=o->ch[1];
    o->ch[1]=null;
    l->maintain();
}
void Print(Node* o){
    if(o!=null){
        o->pushdown();
        Print(o->ch[0]);
        if(o->v!=0)printf("%d ",o->v);
        Print(o->ch[1]);
    }
}
int main(){
    freopen("data.txt","r",stdin);
    scanf("%d%d",&n,&m);
    Build(ro,n+1);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&p,&q);
        Split(ro,p,l,o);
        Split(o,q-p+1,mid,r);
        mid->f^=1;
        ro=Merge(l,Merge(mid,r));
//        Print(ro);
//        printf("\n");
    }
    Print(ro);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章