[bzoj3223]文艺平衡树(splay区间反转模板)

解题关键:splay模板题。

复制代码
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<iostream>
#include<cmath>
using namespace std;
typedef long long ll;
const int N = 100005;

int ch[N][2],par[N],val[N],cnt[N],size[N],rev[N],root,ncnt;
int n,m,x,y;

bool chk(int x){
    return ch[par[x]][1]==x;
}

void pushup(int x){
    size[x]=size[ch[x][0]]+size[ch[x][1]]+cnt[x];
}

void pushdown(int x){
    if(rev[x]){
        swap(ch[x][0],ch[x][1]);//反转就是交换左右子树即可成立。
        rev[ch[x][0]]^=1;
        rev[ch[x][1]]^=1;
        rev[x]=0;
    }
}

void rotate(int x){
    int y=par[x],z=par[y],k=chk(x),w=ch[x][k^1];
    ch[y][k]=w;par[w]=y;
    ch[z][chk(y)]=x;par[x]=z;
    ch[x][k^1]=y;par[y]=x;
    pushup(y);pushup(x);
}

void splay(int x,int goal=0){
    while(par[x]!=goal){
        int y=par[x],z=par[y];
        if(z!=goal){
            if(chk(x)==chk(y)) rotate(y);
            else rotate(x);
        }
        rotate(x);
    }
    if(!goal) root=x;
}

void insert(int x){
    int cur=root,p=0;
    while(cur&&val[cur]!=x){
        p=cur;
        cur=ch[cur][x>val[cur]];
    }
    if(cur){
        cnt[cur]++;
    }else{
        cur=++ncnt;
        if(p) ch[p][x>val[p]]=cur;
        ch[cur][0]=ch[cur][1]=0;
        par[cur]=p;val[cur]=x;
        cnt[cur]=size[cur]=1;
    }
    splay(cur);
}

void find(int x){
    int cur=root;
    if(!cur) return;
    while(ch[cur][x>val[cur]]&&x!=val[cur]){
        cur=ch[cur][x>val[cur]];
    }
    splay(cur);
}
//从1开始计数
int kth(int k){
    k++;
    int cur=root;
    while(1){
        pushdown(cur);
        if(ch[cur][0]&&k<=size[ch[cur][0]]){
            cur=ch[cur][0];
        }else if(k>size[ch[cur][0]]+cnt[cur]){
            k-=size[ch[cur][0]]+cnt[cur];
            cur=ch[cur][1];
        }else{
            return cur;
        }
    }
}

int rnk(int x){
    find(x);
    if(val[root]>=x) return size[ch[root][0]];
    else return size[ch[root][0]]+cnt[root];
}

void reverse(int l,int r){
    int x=kth(l-1),y=kth(r+1);
    splay(x);splay(y,x);
    rev[ch[y][0]]^=1;
}

int pre(int x){
    find(x);
    if(val[root]<x) return root;
    int cur=ch[root][0];
    while(ch[cur][1]) cur=ch[cur][1];
    return cur;
}

int succ(int x) {
    find(x);
    if(val[root]>x) return root;
    int cur=ch[root][1];
    while(ch[cur][0]) cur=ch[cur][0];
    return cur;
}

void output(int x){
    pushdown(x);
    if(ch[x][0]) output(ch[x][0]);
    if(val[x]<=n&&val[x]>=1) printf("%d ",val[x]);
    if(ch[x][1]) output(ch[x][1]);
}

void init(){
    insert(-2e9);
    insert(2e9);
}

int main(){
    init();
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) insert(i);
    while(m--){
        scanf("%d%d",&x,&y);
        reverse(x,y);
    }
    output(root);
    return 0;
}
复制代码

 

posted @   Elpsywk  阅读(155)  评论(0)    收藏  举报
编辑推荐:
· Hangfire Redis 实现秒级定时任务,使用 CQRS 实现动态执行代码
· Android编译时动态插入代码原理与实践
· 解锁.NET 9性能优化黑科技:从内存管理到Web性能的最全指南
· 通过一个DEMO理解MCP(模型上下文协议)的生命周期
· MySQL下200GB大表备份,利用传输表空间解决停服发版表备份问题
阅读排行:
· .NET周刊【4月第1期 2025-04-06】
· 国产的 Java Solon v3.2.0 发布(央企信创的优选)
· centos停服,迁移centos7.3系统到新搭建的openEuler
· 工良出品 | 长文讲解 MCP 和案例实战
· (原创)[开源][.Net Framework 4.0] SimpleLiveDataFeed(极易
点击右上角即可分享
微信分享提示