Splay 支持序列分裂合并模板

//使用每个指针之前都要特别注意是否为空

#include<iostream>
#include<cstring>
#include<set>
#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<deque>
#include<list>
#include<algorithm>
#include<stdio.h>
#include<iomanip>

#define rep(i,n) for(int i=0;i<n;++i)
#define fab(i,a,b) for(int i=a;i<=b;++i)
#define fba(i,b,a) for(int i=b;i>=a;--i)
#define PB push_back
#define INF 0x3f3f3f3f
#define MP make_pair
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define sf scanf
#define pf printf
#define LL long long
const int N=1005;
using namespace std;
typedef pair<int,int>PII;
struct Node{
    Node* ch[2];
    int s,v,flip;
    Node(){
        ch[0]=ch[1]=NULL;
        v=flip=0;s=1;
    }
    Node(int v){
        this->v=v;
        s=1;
        flip=0;
        ch[0]=ch[1]=NULL;
    }
    void maintain(){//节点维护的信息
        s=1;
        if(ch[0]!=NULL)s+=ch[0]->s;
        if(ch[1]!=NULL)s+=ch[1]->s;
    }
    int cmp(int k)const{
        int d= (ch[0]==NULL?0:ch[0]->s);
        if(d+1==k)return -1;
        else return k <= d ? 0:1;
    }
    void pushdown(){//节点信息下传
        if(flip){
            flip=0;
            swap(ch[0],ch[1]);
            if(ch[0]!=NULL)ch[0]->flip^=1;
            if(ch[1]!=NULL)ch[1]->flip^=1;
        }
    }
};
void rotate(Node* &o,int d){//d=0 左旋   1 右旋
    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){//把序列的第k(1<=k<= o->s)个节点伸展为根节点
    o->pushdown();
    int d = o->cmp(k);
    int l=(o->ch[0]==NULL? 0:o->ch[0]->s);
    if(d!=-1){
        if(d==0)splay(o->ch[0],k);
        else splay(o->ch[1],k-l-1);
        rotate(o,d^1);
    }
}
Node *Left,*Right,*Root,*Mid,*Temp;
Node* build(int l,int r){
    if(l==r)return new Node(l);
    else{
        int m=(l+r)>>1;
        Node* t=new Node(m);
        if(l<m)t->ch[0] = build(l,m-1);//记得判断空指针
        t->ch[1]=build(m+1,r);//he he 
        t->maintain();
        return t;
    }
}
void removetree(Node* &o){
    if(o->ch[0]!=NULL)removetree(o->ch[0]);
    if(o->ch[1]!=NULL)removetree(o->ch[1]);
    delete o;
}
void print(Node* o){
    if(o==NULL)return;
    o->pushdown();
    if(o->ch[0]!=NULL)print(o->ch[0]);
    pf("%d\n",o->v);
    if(o->ch[1]!=NULL)print(o->ch[1]);
}
//here left must not be null 
Node* merge(Node* left,Node* right){
    if(left==NULL)return right;
    else if(right==NULL)return left;
    splay(left,left->s);
    left->ch[1]=right;//
    left->maintain();
    return left;
}
//把o节点子树的前k个放在left,剩下的放在right子树
void split(Node* o,int k,Node* &left,Node* &right){
    if(k==0){//特判
        left=NULL;
        right=o;
        return ;
    }
    splay(o,k);
    left=o;
    right=o->ch[1];
    o->ch[1]=NULL;
    left->maintain();
}
int n,m;
int main(){
    while(~sf("%d%d",&n,&m)){
        if(Root!=NULL){
           removetree(Root);
        }
        Root=build(1,n);
        while(m--){
            int a,b;
            sf("%d%d",&a,&b);
            split(Root,a-1,Left,Temp);
            split(Temp,b-a+1,Mid,Right);
            Mid->flip^=1;
            Root = merge( merge(Left,Right),Mid);
        }
        print(Root);
    }
    return 0;
}
//白色上的模板,先静态申请结构体数组,再动态使用,时间应该更快;还有个小技巧,它的空指针用真实的null指针代替,这样即使访问了null的内容也没关系,减少出错的可能性
// UVa11922 Permutation Transformer
// Rujia Liu
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;

struct Node {
  Node *ch[2];
  int s;
  int flip;
  int v;
  int cmp(int k) const {
    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 = !ch[0]->flip;
      ch[1]->flip = !ch[1]->flip;
    }
  }
};

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);
  }
}

// 合并left和right。假定left的所有元素比right小。注意right可以是null,但left不可以
Node* merge(Node* left, Node* right) {
  splay(left, left->s);
  left->ch[1] = right;
  left->maintain();
  return left;
}

// 把o的前k小结点放在left里,其他的放在right里。1<=k<=o->s。当k=o->s时,right=null
void split(Node* o, int k, Node* &left, Node* &right) {
  splay(o, k);
  left = o;
  right = o->ch[1];
  o->ch[1] = null;
  left->maintain();
}

const int maxn = 100000 + 10;
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 = 0;
    null->s = 0;
    root = build(sz);
  }
};

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]);
  }
}

void debug(Node* o) {
  if(o != null) {
    o->pushdown();
    debug(o->ch[0]);
    printf("%d ", o->v-1);
    debug(o->ch[1]);
  }
}

SplaySequence ss;
int main()
{
  int n, m;
  scanf("%d%d", &n, &m);
  ss.init(n+1); // 最前面有一个虚拟结点

  while (m--) {
    int a, b;
    scanf("%d%d", &a, &b);
    Node *left, *mid, *right, *o;
    split(ss.root, a, left, o); // 如果没有虚拟结点,a将改成a-1,违反split的限制
    split(o, b-a+1, mid, right);
    mid->flip ^= 1;
    ss.root = merge(merge(left, right), mid);
  }

  print(ss.root);
  for(int i = 1; i < ans.size(); i++)
    printf("%d\n", ans[i]-1); // 节点编号减1才是本题的元素值

  return 0;
}

 

posted on 2014-10-28 20:31  wanggp3  阅读(304)  评论(0编辑  收藏  举报

导航