平衡樹模板&&bzoj 3223&&Tyvj 1729 文藝平衡樹

Time Limit: 10 Sec Memory Limit: 128 MB
您需要寫一種數據結構(可參考題目標題),來維護一個有序數列,其中需要提供以下操作:翻轉一個區間,例如原有序序列是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

平衡樹的基礎運用。在翻轉區間[l,r]時先將節點l-1,r+1分別轉到head,head->ch[1],此時將標記打在head->ch[1]->ch[0]上即可。
注意在伸展時將伸張路徑上的點都進行一次標記的下傳,以及在統計點數和尋找節點時都要注意標記的下傳。
具體下傳方式就是如果該節點有標記,則將其左右子樹交換位置,並使其子樹的標記異或1。

代碼如下:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#define N 100010
using namespace std;
struct node { int num,mark,f,size;node *ch[2],*fa;node() { num=f=mark=size=0;fa=ch[0]=ch[1]=NULL; } }s[N];
node *null,*head;int n,m,l,r;
struct memory {
    int tot;memory() { tot=0; }
    node *out(int n,int f,node *fa) { node *p=s+tot++;p->num=n,p->f=f,p->fa=fa;return p; }
}pool;
void mark_down(node *now) {
    if(now==NULL) return;
    if(now->mark) {
      swap(now->ch[0],now->ch[1]); now->mark=0;
      if(now->ch[0]!=NULL) now->ch[0]->mark^=1,now->ch[0]->f^=1;
      if(now->ch[1]!=NULL) now->ch[1]->mark^=1,now->ch[1]->f^=1;
    }
    return;
}
void count(node *now) {
    if(now==NULL) return;
    int t=1;
    if(now->ch[0]!=NULL) t+=now->ch[0]->size;
    if(now->ch[1]!=NULL) t+=now->ch[1]->size;
    now->size=t; return;
}
void check(node *now,node *to) {
    node *tr[N];int tot=0;
    while(now!=to) { tr[++tot]=now;now=now->fa; }
    while(tot>0) { mark_down(tr[tot--]); }
}
void rorate(node *now) { 
    node *fa=now->fa;int d=now->f;
    if(now->ch[d^1]!=NULL) now->ch[d^1]->fa=fa,now->ch[d^1]->f=d;fa->ch[d]=now->ch[d^1];
    now->fa=fa->fa;fa->fa->ch[fa->f]=now;now->f=fa->f;fa->f=d^1;
    now->ch[d^1]=fa;fa->fa=now;
    count(fa);count(now); return;
}
void splay(node *now,node *to) {
    if(now==NULL) return;
    check(now,to); while(now->fa!=to) {
      if(now->fa->fa==to) { rorate(now); break; }
      else if(now->f==now->fa->f) rorate(now->fa),rorate(now);
      else rorate(now),rorate(now);
    }if(to==null) head=now;
}
void build(int sum) {
    null=pool.out(0,0,null),head=pool.out(0,0,null);
    sum+=2;head->size=sum; node *now=head;int t=1;sum--;
    while(sum--) {
      now->ch[1]=pool.out(t++,1,now);now=now->ch[1];now->size=n+1;
      if(rand()%2) splay(now,null);
    } return;
}
node *find(node *now,int t,node *to) {
    while(now!=NULL) {
      mark_down(now);int ti;
      if(now->ch[0]==NULL) ti=1;
      else ti=now->ch[0]->size+1;
      if(ti==t) { splay(now,to);return now; }
      else if(ti>t) now=now->ch[0];
      else t-=ti,now=now->ch[1];
    }
}
void rev(int l,int r) {
    if(l>r) swap(l,r);
    node *left=find(head,l,null),*right=find(head,r,head);
    node *now=head->ch[1]->ch[0];
    if(now==NULL) return;else now->mark=now->mark^1;
}
void print(node *now) {
    if(now==NULL) return;mark_down(now);
    print(now->ch[0]);
    if(now->num!=0&&now->num<=n) printf("%d ",now->num);
    print(now->ch[1]);
}
int main() {
    scanf("%d%d",&n,&m);build(n);
    for(int i=1;i<=m;i++)  scanf("%d%d",&l,&r),rev(l,r+2);
    print(head);return 0;
}
發佈了22 篇原創文章 · 獲贊 0 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章