Treap 模板

已过的题目：POJ 2352

/**
**  ch:节点的孩子节点
**  fa:夫节点
**  fix;优先级
**  size：当前子树的大小
**  tot：节点个数（包括sroot）
**  sroot：超级根节点
**  key：节点的键值
*/
const int maxn = 15000 + 50,INF = 0x7fffffff;
int ch[maxn][2],fa[maxn],fix[maxn],size[maxn];
int tot,root,sroot;
int key[maxn];


/**
**  Treap初始化
*/
void Treap_Init(){
    tot = 1;
    root = sroot = 0;
    ch[sroot][0] = ch[sroot][1] = 0;
    fa[sroot] = 0;
    size[sroot] = 0;
    fix[sroot] = key[sroot] = INF;
}

/**
**  新建一个父节点为FA，键值为k的节点，返回节点编号
**
*/
int NewNode(int FA,int k){
    ch[tot][0] = ch[tot][1] = 0;
    fa[tot] = FA;
    size[tot] = 1;
    fix[tot] = rand();
    key[tot] = k;
    return tot++;
}

/**
**  更新rt的size信息
*/
void PushUp(int rt){ size[rt] = size[ch[rt][0]] + size[ch[rt][1]] + 1; if(rt == sroot) size[rt] = 0;}

/**
**  旋转操作
**  kind：   0.表示rt为fa[rt]的左孩子 进行右旋操作 zig
**  kind：   1.表示rt为fa[rt]的右孩子 进行左旋操作 zag
*/
void rotate(int rt,int kind){
    int prt = fa[rt];
    ch[prt][kind] = ch[rt][!kind];fa[ch[rt][!kind]] = prt;
    ch[fa[prt]][ch[fa[prt]][1] == prt] = rt;fa[rt] = fa[prt];
    ch[rt][!kind] = prt;fa[prt] = rt;
    PushUp(prt);PushUp(rt);
}

/**
**  在根为rt的树中插入键值为K的元素 这里可以重复键值
*/
void Insert(int rt,int k){
    while(ch[rt][k >= key[rt]]) rt = ch[rt][k >= key[rt]];
    ch[rt][k >= key[rt]] = NewNode(rt,k);
    rt = ch[rt][k >= key[rt]];
    while(fa[rt] != sroot && fix[rt] > fix[fa[rt]]) rotate(rt,ch[fa[rt]][1] == rt);
    if(fa[rt] == sroot) root = rt;
    while(fa[rt] != sroot){
        rt = fa[rt];PushUp(rt);
    }
}

/**
**  删除键值为k的节点，使用前要保证一定有键值为k的节点
*/
void Delete(int rt,int k){
    if(key[rt] == k){
        if(ch[rt][0] && ch[rt][1]){
            int nrt = ch[rt][0];
            if(fix[nrt] < fix[ch[rt][1]]) nrt = ch[rt][1];
            rotate(nrt,ch[rt][1] == nrt);
            Delete(rt,k);PushUp(nrt);
        }
        else{
            int nrt = ch[rt][0];
            if(nrt == sroot) nrt = ch[rt][1];
            ch[fa[rt]][ch[fa[rt]][1] == rt] = nrt;fa[nrt] = fa[rt];
        }
    }
    else{
        Delete(ch[rt][k >= key[rt]],k);
        PushUp(rt);
    }
}

/**
**  在rt的树中查找键值第k小的节点 返回节点编号
*/
int Search(int rt,int k){
    if(k <= size[ch[rt][0]]) return Search(ch[rt][0],k);
    else if(k > size[ch[rt][0]] + 1) return Search(ch[rt][1],k - 1 - size[ch[rt][0]]);
    else return rt;
}

/**
**  查找键值为k的节点在树中是第几小的
*/
int getRank(int rt,int k){
    if(rt == sroot) return 0;
    if(k >= key[rt]) return size[ch[rt][0]] + 1 + getRank(ch[rt][1],k);
    else return getRank(ch[rt][0],k);
}