SBT模板

最新推荐文章于 2016-08-16 07:19:27 发布

yu836618672

最新推荐文章于 2016-08-16 07:19:27 发布

阅读量698

点赞数

分类专栏：线段树

本文链接：https://blog.youkuaiyun.com/yu836618672/article/details/9899873

版权

线段树专栏收录该内容

2 篇文章

订阅专栏

本文介绍了SBT树（Splay Tree）的相关操作，包括右旋、左旋、维护SBT树性质的函数，以及插入、查找、删除节点的方法。通过这些操作，可以高效地进行数据的增删查改。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<stdio.h>
#include<string.h>
using namespace std;
#define mm 111111
#define mn 33333
int L[mm],R[mm],S[mm],V[mm];
bool p[mm]={0};
int i,x,y,n,a,tt,root,ans;
void RR(int &t)//右旋左节点存在的情况下
{
    int k=L[t];
    L[t]=R[k];
    R[k]=t;
    S[k]=S[t];
    S[t]=S[L[t]]+S[R[t]]+1;
    t=k;
}
void LR(int &t)//左旋
{
    int k=R[t];
    R[t]=L[k];
    L[k]=t;
    S[k]=S[t];
    S[t]=S[L[t]]+S[L[t]]+1;
    t=k;
}
void maintain(int &t,bool flag)//维护sbt树 保证sbt树的性质
{
    if(flag)
        if(S[R[R[t]]]>S[L[t]])
            LR(t);
        else
        if(S[L[R[t]]]>S[L[t]])
            RR(R[t]),LR(t);
        else return;
    else
        if(S[L[L[t]]]>S[R[t]])
            RR(t);
        else if(S[R[L[t]]]>S[R[t]])
            {LR(L[t]);RR(t);}
        else
        return;
    maintain(L[t],0);
    maintain(R[t],1);
    maintain(t,0);
    maintain(t,1);
}
void insert(int &t,int v)//插入
{
    if(t)
    {
        ++S[t];
        if(v<V[t]) insert(L[t],v);
        else insert(R[t],v);
        maintain(t,v>=V[t]);
    }
    else
    {
        V[t=++tt]=v;
        S[t]=1;
        L[t]=R[t]=0;
    }
}
int pred(int t,int v)//在以t为根的子树内比v小的最大值
{
    if(!t) return v;
    if(v<=V[t]) return pred(L[t],v);
    else
    {
        int tmp=pred(R[t],v);
        return tmp==v?V[t]:tmp;
    }
}
int Succ(int t,int v)//在子树内比v大的最小值
{
    if(!t)return v;
    if(v>=V[t])return Succ(R[t],v);
    else
    {
        int tmp=Succ(L[t],v);
        return tmp==v?V[t]:tmp;
    }
}
int Delete(int &t,int v)//删除价值为v的节点
{
    --S[t];
    if(v==V[t]||(v<V[t]&&!L[t])||(v>V[t]&&!R[t]))
    {
        int tmp=V[t];
        if(!L[t]||!R[t])t=L[t]+R[t];
        else V[t]=Delete(L[t],V[t]+1);
        return tmp;
    }
    else if(v<V[t])return Delete(L[t],v);
    else return Delete(R[t],v);
}
int Find(int t,int v)//找到价值为v的节点并返回
{
    while(t&&v!=V[t])
    t=v<V[t]?Find(L[t],v):Find(R[t],v);
    return t;
}
int Rank(int t,int v)//返回根为t的树中键值v的排名。也就是树中键值比v小的结点数+1
{
    if(!t)return 1;
    if(v<=V[t])return Rank(L[t],v);
    else return S[L[t]]+1+Rank(R[t],v);
}
int Select(int t,int k)//返回根为t的树中排名为k的结点。同时该操作能够实现Get-min,Get-max,因为Get-min等于Select(t,1),Get-max等于Select(t,s[t])
{
    if(k==S[L[t]]+1)return V[t];
    if(k<=S[L[t]])return Select(L[t],k);
    else return Select(R[t],k-1-S[L[t]]);
}
void init ()
{
    ans=tt=root=S[0]=0;
}