hdu 4585- HDU 4585 Shaolin Treap + 寻找前驱后继 水

 题意：

每次添加一个数在集合中，并寻找这个数在集合排序后的前驱和后继 

treap模版水题

//treap//插入、删除、查询数排名、查询第k小、查询前驱与后继、查询树大小#include <cstdio>#include <cmath>#include <cstring>#include <string>#include <algorithm>#include <iostream>#include <queue>#include <map>#include <set>#include <vector>using namespace std;#include <iostream>using namespace std;#define MAXN 200010struct  node{    int id,val;    node() {}    node(int a,int b)    {        id=a,val=b;    };    bool operator>(const node &b) const    {        return val>b.val;    }};struct data{    int l,r,size,rnd,w;    node v;};class tp{public:    int n,size,root;    node ans;    tp()    {        n=root=0;        size=0;    }    data tr[MAXN];    void update(int k)//更新结点信息    {        tr[k].size=tr[tr[k].l].size+tr[tr[k].r].size+tr[k].w;    }    void init()    {        n=root=0;        size=0;        memset(tr,0,sizeof tr);    }    void rturn(int &k)    {        int t=tr[k].l;        tr[k].l=tr[t].r;        tr[t].r=k;        tr[t].size=tr[k].size;        update(k);        k=t;    }    void lturn(int &k)    {        int t=tr[k].r;        tr[k].r=tr[t].l;        tr[t].l=k;        tr[t].size=tr[k].size;        update(k);        k=t;    }    void insert(int &k,node x)    {        if(k==0)        {            size++;            k=size;            tr[k].size=tr[k].w=1;            tr[k].v=x;            tr[k].rnd=rand();            return;        }        tr[k].size++;        /*if(tr[k].v==x)tr[k].w++;//每个结点顺便记录下与该节点相同值的数的个数        else */if(x>tr[k].v)        {            insert(tr[k].r,x);            if(tr[tr[k].r].rnd<tr[k].rnd)lturn(k);//维护堆性质        }        else        {            insert(tr[k].l,x);            if(tr[tr[k].l].rnd<tr[k].rnd)rturn(k);        }    }    void del(int &k,node& x)// 删除x数(若有多个相同的数，因只删除一个)    {        if(k==0)return;        /* if(tr[k].v==x)         {             if(tr[k].w>1)             {                 tr[k].w--;                 tr[k].size--;                 return;//若不止相同值的个数有多个，删去一个             }             if(tr[k].l*tr[k].r==0)k=tr[k].l+tr[k].r;//有一个儿子为空             else if(tr[tr[k].l].rnd<tr[tr[k].r].rnd)                 rturn(k),del(k,x);             else lturn(k),del(k,x);         }         else*/ if(x>tr[k].v)            tr[k].size--,del(tr[k].r,x);        else tr[k].size--,del(tr[k].l,x);    }    void query_pro(int k,node& x)//求x的前驱(前驱定义为小于x，且最大的数)    {        if(k==0)return;        if(x>tr[k].v)        {            ans=tr[k].v;            query_pro(tr[k].r,x);        }        else query_pro(tr[k].l,x);    }    void query_sub(int k,node& x)//求x的后继(后继定义为大于x，且最小的数)    {        if(k==0)return;        if(tr[k].v>x)        {            ans=tr[k].v;            query_sub(tr[k].l,x);        }        else query_sub(tr[k].r,x);    }};tp sb;int main(){    int n;    int a,b;     while(scanf("%d", &n)!=EOF)    {        if (!n)break;        sb.init();        sb.insert(sb.root,node(0,-1e9));        sb.insert(sb.root,node(1,1e9));        node tmp;        for(int   i = 1; i <= n; i++)        {            scanf("%d%d",&a,&b);            tmp.id=a,tmp.val=b;            sb.insert(sb.root,tmp);            sb.query_pro(sb.root,tmp);            node up=sb.ans;            sb.query_sub(sb.root,tmp);            node dn=sb.ans;            int dif1=abs(up.val-b);            int dif2=abs(dn.val-b);            if (dif1<=dif2)                printf("%d %d\n",a,up.id);            else if (dif1>dif2)                printf("%d %d\n",a,dn.id);        }    }    return 0;}