BZOJ3224 CODEVS4543 普通平衡树 题解&代码

醉啦醉啦= =第k大第k小纠结了好久，最后瞎蒙了一个蒙错了然后纠结了好久，后来学长告诉我【你们排名难道是成绩低的排前面么】

题意：
维护一个序列，按照val[]排序，支持：
1. 插入x
2. 删除x(若有多个相同的val，因只删除一个)
3. 查询x是第几大(若有多个相同的数，因输出最小的排名)
4. 查询第k大
5. 求x的前驱(前驱定义为小于x，且最大的数，可能树中不存在x)
6. 求x的后继(后继定义为大于x，且最小的数，同上)

思路：
维护一个顺序序列，维护s[]表示节点子树大小，z[]表示节点个数
求前驱后继的时候先插入一个再删除就行

#include<iostream>
#include<stdio.h>
#define lson ch[x][0]
#define rson ch[x][1]
using namespace std;
const int maxn=100005;
int n,op,x,tot,root,fa[maxn],ch[maxn][2],val[maxn],s[maxn],z[maxn];
void link(int x,int y,int d)
{
    if(y)ch[y][d]=x;
    if(x)fa[x]=y;
}
void maintain(int x)
{
    if(x)s[x]=s[lson]+s[rson]+z[x];
}
int ischild(int x)
{
    return ch[fa[x]][1]==x;
}
void rotate(int x)
{
    int y=fa[x],d=ischild(x);
    if(!fa[y])root=x;
    link(x,fa[y],ischild(y));
    link(ch[x][!d],y,d);
    link(y,x,!d);
    maintain(y);
}
void Splay(int x,int goal=0)
{
    int y=fa[x],z=fa[y];
    while(y!=goal)
    {
        if(z==goal)
        {
            rotate(x);
            break;
        }
        if(ischild(x)^ischild(y))rotate(x);
        else rotate(y);
        rotate(x);
        y=fa[x],z=fa[y];
    }
    maintain(x);
}
int newnode(int v)
{
    int t=++tot;
    val[t]=v;
    s[t]=z[t]=1;
    return t;
}
int getval(int value)
{
    int x=root;
    while(x)
    {
        if(val[x]==value)return x;
        if(ch[x][val[x]<value])x=ch[x][val[x]<value];
        else return x;
    }
    return x;
}
void addtree(int v)
{
    if(!root)
    {
        root=newnode(v);
        return;
    }
    int x=getval(v),t=x;
    if(val[x]==v)z[x]++;
    else t=newnode(v),link(t,x,val[x]<v);
    Splay(t);
}
void deltree(int v)
{
    int x=getval(v);
    if(z[x]>1)
    {
        z[x]--;
        Splay(x);
        return;
    }
    Splay(x);
    x=ch[x][0];
    while(ch[x][1])x=ch[x][1];
    if(x)
    {
        Splay(x,root);
        link(ch[root][1],x,1);
        root=x;
        fa[root]=0;//false
    }
    else
    {
        root=ch[root][1];
        fa[root]=0;
    }
    maintain(root);
}
int getvalth(int v)
{
    addtree(v);
    int x=getval(v);
    Splay(x);
    int ans=s[lson]+1;
    deltree(v);
    return ans;
}
int getkth(int k)
{
    //cout<<k<<' '<<x<<endl;
    int x=root;
    while(x)
    {
        //cout<<ch[x][0]<<' '<<z[x]<<endl;
        if(k<=s[ch[x][0]])x=ch[x][0];
        else if(k<=s[ch[x][0]]+z[x])return val[x];
                else k-=s[ch[x][0]]+z[x],x=ch[x][1];
    }
    //cout<<x<<endl;
    return val[x];
}
int getpre(int v)
{
    addtree(v);
    int x=ch[root][0];
    while(ch[x][1])x=ch[x][1];
    deltree(v);
    return val[x];
}
int getnext(int v)
{
    addtree(v);
    int x=ch[root][1];
    while(ch[x][0])x=ch[x][0];
    deltree(v);
    return val[x];
}
void print(int x)
{
    if(ch[x][0])print(ch[x][0]);
    printf("%d %d\n",x,val[x]);
    if(ch[x][1])print(ch[x][1]);
}
int main(void)
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&op,&x);
        //cout<<op<<endl;
        if(op==1)
            addtree(x);
        if(op==2)
            deltree(x);
        if(op==3)
            printf("%d\n",getvalth(x));
        if(op==4)
            printf("%d\n",getkth(x));
        if(op==5)
            printf("%d\n",getpre(x));
        if(op==6)
            printf("%d\n",getnext(x));
        //print(root);
    }
    return 0;
}