bzoj3224(treap)

3224: Tyvj 1728 普通平衡树

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 1844  Solved: 727
[Submit][Status]

Description

您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:
1. 插入x数
2. 删除x数(若有多个相同的数,因只删除一个)
3. 查询x数的排名(若有多个相同的数,因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x,且最大的数)
6. 求x的后继(后继定义为大于x,且最小的数)

Input

第一行为n,表示操作的个数,下面n行每行有两个数opt和x,opt表示操作的序号(1<=opt<=6)

Output

对于操作3,4,5,6每行输出一个数,表示对应答案

Sample Input

10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598

Sample Output

106465
84185
492737

HINT

1.n的数据范围:n<=100000

2.每个数的数据范围:[-1e7,1e7]

题意:RT

思路:平衡树裸题,我用treap实现的,由于敲rot的时候少敲了一个字符导致无限RE,sad~,还是得小心的敲哎,跑完412ms

            这题有个坑,对于操作3,要找出最小排名,比如现在有1 3 3 5   那么在输入3 3以后就要输出2才对,其它的也没啥要注意的了,放开写就行了~

#include 
#include 
#include 
#include 
using namespace std;
#define INF 1000000000

struct node{
    int v;
    int s;
    int sz;
    node *p;
    node *ch[2];

    node(){
        v=sz=0;
        s=-1;
        p=ch[0]=ch[1]=this;
    }

    bool cmp(int s)
    {
        return this->sch[1]==o?1:0;
    }
}Tnull,*null=&Tnull;

void newnode(node *pre,int v,int d)
{
    node *u=(node*)malloc(sizeof(node));
    u->ch[0]=u->ch[1]=null;
    u->sz=1;
    u->s=rand();
    u->v=v;
    u->p=pre;
    pre->ch[d]=u;
}

void pushup(node *o)
{
    o->sz=o->ch[0]->sz+o->ch[1]->sz+1;
}

void rot(node *o,int d)
{
    node *k=o->ch[d];
    o->ch[d]=k->ch[d^1];
    k->ch[d^1]->p=o;
    o->p->ch[o->p->cd(o)]=k;
    k->p=o->p;
    k->ch[d^1]=o;
    o->p=k;
    pushup(o);
    pushup(k);
}

void Insert(node *o,node *pre,int v,int dd)
{
    if(o==null){
        newnode(pre,v,dd);
        return;
    }
    int d;
    if(v>=o->v){
        d=1;
        Insert(o->ch[1],o,v,1);
    }
    else {
        d=0;
        Insert(o->ch[0],o,v,0);
    }
    if(o->cmp(o->ch[d]->s))rot(o,d);
    else pushup(o);
}

void Delete(node *o,int v)
{
    if(o==null)return;
    if(o->v==v){

        while(o->ch[0]!=null&&o->ch[1]!=null)
        {
            int d=o->ch[0]->cmp(o->ch[1]->s);
            rot(o,d);
            --o->p->sz;
        }
        node *k;
        o->ch[0]==null ? k=o->ch[1] : k=o->ch[0];
        o->p->ch[o->p->cd(o)]=k;
        k->p=o->p;
        free(o);
        return;
    }
    v>o->v? Delete(o->ch[1],v) : Delete(o->ch[0],v);
    pushup(o);
}

int Find(node *o,int v)
{
    if(o==null)return 1;
    int ans=0;
    if(v<=o->v)ans=Find(o->ch[0],v);
    else ans=Find(o->ch[1],v)+o->ch[0]->sz+1;
    return ans;
}

int Find2(node *o,int k)
{
    if(o==null)return -1;
    if(o->ch[0]->sz+1==k)return o->v;
    return o->ch[0]->sz>=k? Find2(o->ch[0],k) : Find2(o->ch[1],k-1-o->ch[0]->sz) ;
}

int mini;
int maxi;
void Findp(node *o,int v)
{
    if(o==null)return;
    if(o->vv)maxi=o->v;
        Findp(o->ch[1],v);
    }
    else Findp(o->ch[0],v);
}

void Findh(node *o,int v)
{
    if(o==null)return;
    if(o->v>v){
        if(mini>o->v)mini=o->v;
        Findh(o->ch[0],v);
    }
    else Findh(o->ch[1],v);
}

void Free(node *o)
{
    if(o==null)return;
    Free(o->ch[0]);
    Free(o->ch[1]);
    free(o);
}

int main()
{
    int n;

    while(scanf("%d",&n)!=EOF)
    {
        node *root=null;
        for(int i=0;ich[1],root,v,1);

            else if(vis==2)Delete(root->ch[1],v);

            else if(vis==3)printf("%d\n",Find(root->ch[1],v));

            else if(vis==4)printf("%d\n",Find2(root->ch[1],v));

            else if(vis==5){
                maxi=-INF;
                Findp(root->ch[1],v);
                printf("%d\n",maxi);
            }
            else {
                mini=INF;
                Findh(root->ch[1],v);
                printf("%d\n",mini);
            }
        }

        Free(root->ch[1]);
        root->ch[1]=null;
    }

    return 0;
}
阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页