splay版

指针是个好东西

不过就是得判空

还有别忘传引用（其实应该都传引用）

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
int inf=0x7fffffff;
struct node* nil;
struct node
{
    int num;
    int val;
    int size;
    node* ch[2];
    node (int v) :val(v)
    {
        size=1;
        num=1;
        ch[0]=ch[1]=nil;
    }
    void sum()
    {
        size=num;
        if(ch[0]!=nil)
            size+=ch[0]->size;
        if(ch[1]!=nil)
            size+=ch[1]->size;
        return ;
    }
    int cmp(int v)
    {
        if(val==v)
            return -1;
        return (val>v ? 0 : 1);
    }
    int cmpkth(int k)
    {
        int s=( ch[0]==nil ? 0 : ch[0]->size );
        if(k>s&&k<=s+num)
            return -1;
        if(k<=s)
            return 0;
        else
            return 1;
    }
};
node* root;
void visit(node* x)
{
    if(x==nil)
        return ;
    visit(x->ch[0]);
    printf("%d ",x->val);
    visit(x->ch[1]);
    return ;
}
void rotato(node* &x,int base)
{
    node* k=x->ch[base^1];
    x->ch[base^1]=k->ch[base];
    k->ch[base]=x;
    x->sum();
    k->sum();
    x=k;
}
void splay(node* &x,int v)
{
    int d=x->cmp(v);
    if(d!=-1&&x->ch[d]!=nil)
    {
        int d2=x->ch[d]->cmp(v);
        if(d2!=-1&&x->ch[d]->ch[d2]!=nil)
        {
            splay(x->ch[d]->ch[d2],v);
            if(d==d2)
                rotato(x,d2^1),rotato(x,d^1);
            else
                rotato(x->ch[d],d2^1),rotato(x,d^1);
        }
        else
            rotato(x,d^1);
    }
}
void splaykth(node* &x,int k)
{
    int d=x->cmpkth(k);
    if(d!=-1)
    {
        if(d==1)
            k=k-x->ch[0]->size-x->num;
        int d2=x->ch[d]->cmpkth(k);
        if(d2!=-1)
        {
            int k2=(d2==1 ? k-x->ch[d]->ch[0]->size-x->ch[d]->num : k);
            splaykth(x->ch[d]->ch[d2],k2);
            if(d==d2)
                rotato(x,d2^1),rotato(x,d^1);
            else
                rotato(x->ch[d],d2^1),rotato(x,d^1);
        }
        else
            rotato(x,d^1);
    }
    return ;
}
void pre(node* x,int val,int &ans)
{
    if(x==nil)
        return ;
    if(x->val<val)
    {
        if(x->val>ans)
            ans=x->val;
        if(x->ch[1]!=nil)
            pre(x->ch[1],val,ans);
    }
    else
        if(x->val>=val&&x->ch[0]!=nil)
            pre(x->ch[0],val,ans);
}
void nxt(node* x,int val,int &ans)
{
    if(x==nil)
        return ;
    if(x->val>val)
    {
        if(x->val<ans)
            ans=x->val;
        if(x->ch[0]!=nil)
            nxt(x->ch[0],val,ans);
    }
    else
        if(x->val<=val&&x->ch[1]!=nil)
            nxt(x->ch[1],val,ans);
}
int find(node* &x,int val)
{
    splay(x,val);
    return x->ch[0]->size+1;
}
int kth(node* &x,int k)
{
    splaykth(x,k);
    return x->val;
}
node *spilt(node* &x,int val)
{
    if(x==nil)
        return nil;
    splay(x,val);
    node* t1;
    node* t2;
    if(x->val<=val)
        t1=x,t2=x->ch[1],t1->ch[1]=nil;
    else
        t2=x,t1=x->ch[0],t2->ch[0]=nil;
    x->sum();
    x=t1;
    return t2;
}
void merge(node* &t1,node* &t2)
{
    if(t1==nil)
        swap(t1,t2);
    splay(t1,inf);
    t1->ch[1]=t2;
    t2=nil;
    t1->sum();
}
void insert(node* &x,int val)
{
    //visit(x);printf("\n");
    node* t2=spilt(x,val);
    //visit(x);printf("\n");
    if(x!=nil&&x->val==val)
    {
        x->num+=1;
        x->sum();
    }
    else
    {
        node* nw=new node(val);
        merge(x,nw);
    }
    merge(x,t2);
    //visit(root);printf("\n");
}
void erase(node* &x,int val)
{
    node* t2=spilt(x,val);
    x->num-=1;
    if(x->num==0)
    {
        node* t3=x;
        x=x->ch[0];
        delete t3;
    }
    merge(x,t2);
}
int read()
{
    int s=0,f=1;
    char in=getchar();
    while(in<'0'||in>'9')
    {
        if(in=='-')
            f=-1;
        in=getchar();
    }
    while(in>='0'&&in<='9')
    {
        s=(s<<1)+(s<<3)+in-'0';
        in=getchar();
    }
    return s*f;
}
int main()
{
    int n=read();
    int a,b;
    int ans;
    nil=new node(0);
    root=nil->ch[0]=nil->ch[1]=nil;
    nil->size=nil->num=0;
    for(int i=1;i<=n;i++)
    {
        a=read();
        b=read();
        switch(a)
        {
            case 1: insert(root,b);break;
            case 2: erase(root,b);break;
            case 3: printf("%d\n",find(root,b));break;
            case 4: printf("%d\n",kth(root,b));break;
            case 5: insert(root,b);ans=-0x7fffffff;pre(root,b,ans);printf("%d\n",ans);erase(root,b);break;
            case 6: insert(root,b);ans=0x7fffffff;nxt(root,b,ans);printf("%d\n",ans);erase(root,b);break;
            default: break;
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/Lance1ot/p/8947372.html