二叉搜索树（BST）模版

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1024;
struct node{
    int val;  // val：该节点储存的数值 
    int lch, rch; // lch: 左子树根节点编号，rch：右子树根节点编号
    int cnt; //该节点储存的数出现的次数(由于我们这里维护的是可重集)
    int sz; //以该节点为根节点的子树的节点数目（即子树大小）
};
node bst[MAXN];
int id=1;
//插入 
void insert(int v, int pos=1)//插入
{
    bst[pos].sz++;// 以该节点为根节点的子树大小+1
    if(bst[pos].val==0 && bst[pos].lch==0 && bst[pos].rch==0){//空节点 
        bst[pos].val=v;
        bst[pos].cnt=1;
    }
    else if(v<bst[pos].val){// 向左搜索
        if(bst[pos].lch==0)// 如果应该向左搜，但不存在左节点，则创建一个新节点
            bst[pos].lch=++id;
        insert(v, bst[pos].lch);
    }
    else if(v>bst[pos].val){// 向右搜索
        if(bst[pos].rch==0)
            bst[pos].rch=++id;
        insert(v, bst[pos].rch);
    }
    else//已经存在相同值的点 
        bst[pos].cnt++;
} 
//删除
void remove(int v, int pos=1){
    bst[pos].sz--;
    if(v<bst[pos].val)
        remove(v, bst[pos].lch);
    else if(v>bst[pos].val)
        remove(v, bst[pos].rch);
    else
        bst[pos].cnt--;
} 
//求排名:因为排名被定义为比某数小的数+1，所以我们直接实现两个函数countl和countg，
//用来求比某数小的数的数量和比某数大的数的数量。（这两个函数后面也会用到）
int countl(int v, int pos=1){// 求比某数小的数的个数
    if(v<bst[pos].val)
        return bst[pos].lch?countl(v, bst[pos].lch):0;
    else if(v>bst[pos].val)
        return bst[bst[pos].lch].sz+bst[pos].cnt+bst[pos].rch?countl(v, bst[pos].rch):0;
    else
        return bst[bst[pos].lch].sz;
}
int countg(int v, int pos=1){// 求比某数大的数的个数
    if(v>bst[pos].val)
        return bst[pos].rch?countg(v, bst[pos].rch):0;
    else if(v<bst[pos].val)
        return bst[bst[pos].rch].sz+bst[pos].cnt+bst[pos].lch?countg(v, bst[pos].lch):0;
    else
        return bst[bst[pos].rch].sz;
}
//查询某值在数列中排名 
int rank(int v){
    return countl(v, 1)+1;
}
//求指定排名的数
int kth(int k, int pos=1)// 求指定排名的数
{
    if(bst[bst[pos].lch].sz+1>k)// 答案在左，在左子树中找排名为k的数
        return kth(k, bst[pos].lch);
    else if(bst[bst[pos].lch].sz+bst[pos].cnt<k)// 答案在右，在右子树中找排名为k - size[L[pos]] - N[pos]的数
        return kth(k-bst[bst[pos].lch].sz-bst[pos].cnt, bst[pos].rch);
    else 
        return bst[pos].val;
}
//注意，假如某个数的排名为2，且它出现了3次，那么这个函数传入2、3、4都会返回这个数，这也提供了一些方便。


//求前驱:根据我们kth函数的性质，直接找到排名比当前数小1的那个数即可。
int pre(int v){
    int r=countl(v);
    return kth(r);
}

//求后继:后继的排名则是小于等于当前数的数的数量+1。
int suc(int v){
    int r = bst[1].sz - countg(v) + 1;
    return kth(r);
} 
int main()
{
    //依次插入  10 6 2 5 1 7 4 2 5 9 5
    int n;
    cin>>n;
    for(int i=0; i<n; i++){
        int x;
        cin>>x;
        insert(x);
    }
    for(int i=1; i<=id; i++)
        cout<<i<<":"<<bst[i].val<<" "<<bst[i].lch<<" "<<bst[i].rch<<" "<<bst[i].sz<<" "<<bst[i].cnt<<endl;
    cout<<rank(6)<<endl; 
    return 0;
 }