Splay 模板/学习(待更新

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/IDrandom/article/details/82499897

http://hihocoder.com/contest/hiho104
以hiho104为例
看着hiho里的说明,写了一个这个东西,先上一个好理解的模板

#include<bits/stdc++.h>
#include<unordered_set>
#include<unordered_map>
using namespace std;
typedef long long LL;
typedef set<int>::iterator sit;
const LL mod=1e9+7;
const double eps=1e-9;
const int INF=0x3f3f3f3f;
struct Stree{//空节点统一指向0
    int fa;
    int val;
    int l,r;
    int siz;
}sp[200010];
int root,cnt;

void pushup(int rt){
    sp[rt].siz=sp[sp[rt].l].siz+sp[sp[rt].r].siz+1;
}
void rro(int rt){
    int fa=sp[rt].fa;
    int rs=sp[rt].r;
    int gf=sp[fa].fa;
    sp[fa].l=rs;
    sp[rs].fa=fa;
    if(gf){
        if(sp[gf].l==fa)sp[gf].l=rt;
        else sp[gf].r=rt;
    }else root=rt;
    sp[rt].fa=gf;
    sp[rt].r=fa;
    sp[fa].fa=rt;
    pushup(fa);pushup(rt);
}
void lro(int rt){
    int fa=sp[rt].fa;
    int ls=sp[rt].l;
    int gf=sp[fa].fa;
    sp[fa].r=ls;
    sp[ls].fa=fa;
    if(gf){
        if(sp[gf].l==fa)sp[gf].l=rt;
        else sp[gf].r=rt;
    }else root=rt;
    sp[rt].fa=gf;
    sp[rt].l=fa;
    sp[fa].fa=rt;
    pushup(fa);pushup(rt);
}
void splay(int x,int y){//把编号为x的节点转到y的儿子节点
    while(sp[x].fa!=y){
        int fa=sp[x].fa;
        if(sp[fa].fa==y){
            if(sp[fa].l==x)rro(x);
            else lro(x);
        }
        else {
            int gf=sp[fa].fa;
            if(sp[fa].l==x&&sp[gf].l==fa){rro(fa);rro(x);}
            else if(sp[fa].r==x&&sp[gf].r==fa){lro(fa);lro(x);}
            else if(sp[fa].r==x&&sp[gf].l==fa){lro(x);rro(x);}
            else {rro(x);lro(x);}
        }
    }
}
void ist(int &k,int x,int fa){
    if(!k){
        k=++cnt;
        sp[k].val=x;
        sp[k].fa=fa;
        sp[k].siz=1;
        splay(k,0);
        return ;
    }
    if(x<sp[k].val)ist(sp[k].l,x,k);
    else ist(sp[k].r,x,k);
}
int pre,nxt;
void fdpre(int rt,int x){//找到第一个小于等于x的数,并保存在pre中
    if(!rt)return ;
    //printf("%d %d\n",rt,sp[rt].val);
    if(sp[rt].val==x){pre=rt;return ;}
    if(sp[rt].val<x){pre=rt;fdpre(sp[rt].r,x);}
    else {fdpre(sp[rt].l,x);}
}
void fdnxt(int rt,int x){//找到第一个大于等于x的数,并保存在nxt中
    if(!rt)return ;
    if(sp[rt].val==x){nxt=rt;return ;}
    else if(sp[rt].val>x){nxt=rt;fdnxt(sp[rt].l,x);}
    else {fdnxt(sp[rt].r,x);}
}
void del(int l,int r){//删除区间[l,r]
    fdpre(root,l-1);fdnxt(root,r+1);
    splay(pre,0);splay(nxt,pre);
    sp[nxt].l=0;
    pushup(nxt);pushup(pre);
}
void init(){
    cnt=0;root=0;
    ist(root,-INF,0);
    ist(root,INF,0);
}
void print(int rt){
    if(!rt)return ;
    print(sp[rt].l);
    printf("%d %d %d %d\n",rt,sp[rt].val,sp[rt].l,sp[rt].r);
    print(sp[rt].r);
}
char op[2];
int main(void)
{
    int n,k,a,b;
    while(~scanf("%d",&n)){
        init();
        while(n--){
            scanf("%s",op);
            if(op[0]=='I'){
                scanf("%d",&k);
                ist(root,k,0);
                //print(root);
            }
            else if(op[0]=='Q'){
                scanf("%d",&k);
                fdpre(root,k);
                printf("%d\n",sp[pre].val);
                splay(pre,0);
            }
            else {
                scanf("%d%d",&a,&b);
                del(a,b);
            }
            //for(int i=1;i<=7;i++)printf("%d %d %d %d\n",i,sp[i].val,sp[i].l,sp[i].r);
        }
    }
    return 0;
}

果然还是显得太长了,看一看大佬们的写法,再学习一下,打算再压一下
https://blog.csdn.net/qq_30974369/article/details/77587168
看一下这个大佬的讲解
updated:运行效率上没有任何区别,主要是改掉了rotate函数,然后顺便改下面的函数的时候发现很多东西可以省略掉

#include<bits/stdc++.h>
#include<unordered_set>
#include<unordered_map>
using namespace std;
typedef long long LL;
typedef set<int>::iterator sit;
const LL mod=1e9+7;
const double eps=1e-9;
const int INF=0x3f3f3f3f;
struct Stree{//空节点统一指向0
    int fa,ch[2],val;
    int siz;
}sp[200010];
int root,cnt;//根,节点个数
void pushup(int rt){
    sp[rt].siz=sp[sp[rt].ch[0]].siz+sp[sp[rt].ch[1]].siz+1;
}
void ro(int rt){
    int fa=sp[rt].fa;
    int gf=sp[fa].fa;
    int sid=sp[fa].ch[1]==rt;
    sp[sp[gf].ch[sp[gf].ch[1]==fa]=rt].fa=gf;
    sp[sp[fa].ch[sid]=sp[rt].ch[sid^1]].fa=fa;
    sp[sp[rt].ch[sid^1]=fa].fa=rt;
    if(!gf)root=rt;
    pushup(fa);pushup(rt);
}
void splay(int x,int y){//把编号为x的节点转到y的儿子节点
    while(sp[x].fa!=y){
        int fa=sp[x].fa;
        int gf=sp[fa].fa;
        if(gf!=y){
            if((sp[gf].ch[1]==fa)^(sp[fa].ch[1]==x))ro(x);
            else ro(fa);
        }
        ro(x);
    }
}
void ist(int &k,int x,int fa){
    if(!k){
        k=++cnt;
        sp[k].val=x;sp[k].fa=fa;
        sp[k].siz=1;splay(k,0);
        return ;
    }
    if(x<sp[k].val)ist(sp[k].ch[0],x,k);
    else ist(sp[k].ch[1],x,k);
}
int pre,nxt;
void fdpre(int rt,int x){//找到第一个小于等于x的数,并保存在pre中
    if(!rt)return ;
    //printf("%d %d\n",rt,sp[rt].val);
    if(sp[rt].val==x){pre=rt;return ;}
    if(sp[rt].val<x){pre=rt;fdpre(sp[rt].ch[1],x);}
    else {fdpre(sp[rt].ch[0],x);}
}
void fdnxt(int rt,int x){//找到第一个大于等于x的数,并保存在nxt中
    if(!rt)return ;
    if(sp[rt].val==x){nxt=rt;return ;}
    else if(sp[rt].val>x){nxt=rt;fdnxt(sp[rt].ch[0],x);}
    else {fdnxt(sp[rt].ch[1],x);}
}
void del(int l,int r){//删除区间[l,r]
    fdpre(root,l-1);fdnxt(root,r+1);
    splay(pre,0);splay(nxt,pre);
    sp[nxt].ch[0]=0;
    pushup(nxt);pushup(pre);
}
void init(){
    cnt=0;root=0;
    ist(root,-INF,0);
    ist(root,INF,0);
}
void print(int rt){
    if(!rt)return ;
    print(sp[rt].ch[0]);
    printf("%d %d %d %d\n",rt,sp[rt].val,sp[rt].ch[0],sp[rt].ch[1]);
    print(sp[rt].ch[1]);
}
char op[2];
int main(void)
{
    int n,k,a,b;
    while(~scanf("%d",&n)){
        init();
        while(n--){
            scanf("%s",op);
            if(op[0]=='I'){
                scanf("%d",&k);
                ist(root,k,0);
                //print(root);
            }
            else if(op[0]=='Q'){
                scanf("%d",&k);
                fdpre(root,k);
                printf("%d\n",sp[pre].val);
                splay(pre,0);
            }
            else {
                scanf("%d%d",&a,&b);
                del(a,b);
            }
//          printf("%d\n",root);
//          for(int i=1;i<=7;i++)printf("%d %d %d %d\n",i,sp[i].val,sp[i].ch[0],sp[i].ch[1]);
        }
    }
    return 0;
}

在接下来就是更新用法了。。。挖个坑吧,慢慢

猜你喜欢

转载自blog.csdn.net/IDrandom/article/details/82499897